Uso de ai.embed con pandas

La ai.embed función convierte el texto en incrustaciones vectoriales que representan el significado. Use incrustaciones para buscar, agrupar y comparar contenido mediante el significado en lugar de la redacción exacta.

Nota:

Información general

La función ai.embed extiende la clase de la serie de pandas.

Para generar incrustaciones vectoriales de cada fila de entrada, llame a la función en una serie pandas o en una columna de texto de Pandas DataFrame.

La función devuelve una serie pandas que contiene incrustaciones, que se pueden almacenar en una nueva columna DataFrame.

Syntax

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

Parámetros

Ninguno.

Devoluciones

La función devuelve una Serie de pandas que contiene, para cada fila de entrada, incrustaciones como array de NumPy de valores de coma flotante de 32 bits. El tamaño de la matriz depende de las dimensiones del modelo de inserción, que se pueden configurar en las funciones 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)

Output:

Captura de pantalla de una trama de datos con columnas