Compartilhar via


Usar ai.embed com pandas

A ai.embed função usa IA generativa para converter texto em inserções de vetor. Esses vetores permitem que a IA entenda as relações entre textos, para que você possa pesquisar, agrupar e comparar conteúdo com base no significado em vez da redação exata. Com uma única linha de código, você pode gerar inserções de vetor de uma coluna em um DataFrame.

Observação

Visão geral

A função ai.embed estende a classe pandas Series.

Para gerar inserções de vetor de cada linha de entrada, chame a função em uma série pandas ou uma coluna de texto do Pandas DataFrame.

A função retorna uma série pandas que contém inserções, que podem ser armazenadas em uma nova coluna DataFrame.

Sintaxe

df["embed"] = df["col1"].ai.embed()

Parâmetros

Nenhum.

Devoluções

A função retorna uma série pandas que contém inserções como matriz numpy de float-32 para cada linha de texto de entrada. O número de elementos na matriz depende das dimensões do modelo de inserção, que são configuráveis em funções de IA

Example

# This code uses AI. Always review output for mistakes.

df = pd.DataFrame([
        "This duvet, lovingly hand-crafted from all-natural fabric, is perfect for a good night's sleep.",
        "Tired of friends judging your baking? With these handy-dandy measuring cups, you'll create culinary delights.",
        "Enjoy this *BRAND NEW CAR!* A compact SUV perfect for the professional commuter!"
    ], columns=["descriptions"])
    
df["embed"] = df["descriptions"].ai.embed()
display(df)

Esta célula de código de exemplo fornece a seguinte saída:

Captura de tela de um quadro de dados com as colunas 'descriptions' e 'inserção'. A coluna 'inserção' contém inserções para as descrições.