Usar ai.embed com pandas

A ai.embed função converte texto em inserções de vetor que representam significado. Use embeddings para pesquisar, agrupar e comparar conteúdo com base no significado, em vez da redação exata.

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 valores float-32 para cada linha de entrada. O tamanho da 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)

Saída:

Conteúdo relacionado

• Use ai.embed com o PySpark.
• Saiba mais sobre o AI Functions.
• Alterar a configuração padrão para AI Functions with pandas.
• Entenda a cobrança do AI Functions.