Compartir a través de


Uso de ai.classify con pandas

La ai.classify función usa ia generativa para clasificar el texto de entrada según las etiquetas personalizadas que elija, con una sola línea de código.

Nota:

Información general

La función ai.classify extiende la clase de la serie de pandas. Para asignar etiquetas proporcionadas por el usuario a cada fila de entrada, llame a la función en una columna de texto de un DataFrame de Pandas.

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

Sugerencia

Se recomienda usar la función ai.classify con al menos dos etiquetas de entrada.

Syntax

df["classification"] = df["input"].ai.classify("category1", "category2", "category3")

Parámetros

Nombre Description
labels
Obligatorio
Una o varias cadenas que representan el conjunto de etiquetas de clasificación para que coincidan con los valores de texto de entrada.

Devoluciones

La función devuelve una serie de pandas que contiene una etiqueta de clasificación para cada fila de texto de entrada. Si no se puede clasificar un valor de texto, la etiqueta correspondiente es null.

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["category"] = df['descriptions'].ai.classify("kitchen", "bedroom", "garage", "other")
display(df)

Esta celda de código de ejemplo proporciona la siguiente salida:

Captura de pantalla de una trama de datos con columnas