Use ai.embed com pandas

A ai.embed função converte texto em embeddings vetoriais que representam significado. Use embeddings para pesquisar, agrupar e comparar conteúdos por significado em vez de por palavras exatas.

Visão geral

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

Para gerar embeddings vetoriais de cada linha de entrada, chamar a função numa série pandas ou numa coluna de texto de um DataFrame pandas.

A função devolve uma Série Pandas que contém embeddings, que podem ser armazenadas numa nova coluna do dataframe.

Sintaxe

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

Parâmetros

Nenhum.

Devoluções

A função devolve uma Series do pandas que contém embeddings como um array NumPy de valores float-32 para cada linha de entrada. O tamanho do array depende das dimensões do modelo de embedding, que são configuráveis em AI Functions.

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

• Usa ai.embed com o PySpark.
• Saiba mais sobre Funções de IA.
• Alterar a configuração padrão das Funções de IA com pandas.
• Compreender a faturação das funções de IA.