Verwenden Sie ai.embed mit Pandas

Die ai.embed Funktion konvertiert Text in Vektoreinbettungen, die Bedeutung darstellen. Verwenden Sie Embeddings, um Inhalte nach ihrer Bedeutung statt nach exakter Formulierung zu suchen, zu gruppieren und zu vergleichen.

Hinweis

Überblick

Die ai.embed-Funktion erweitert die Klasse der Pandas-Serie.

Um Vektoreinbettungen jeder Eingabezeile zu generieren, rufen Sie die Funktion entweder für eine Pandas-Serie oder eine Textspalte von Pandas DataFrame auf.

Die Funktion gibt eine pandas-Serie zurück, die Einbettungen enthält. Diese können in einer neuen DataFrame-Spalte gespeichert werden.

Syntax

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

Die Parameter

Keiner.

Rückkehr

Die Funktion gibt eine pandas-Serie zurück, die für jede Eingabezeile Einbettungen als NumPy-Array mit Float32-Werten enthält. Die Arraygröße hängt von den Einbettungsmodellabmessungen ab, die in KI-Funktionen konfigurierbar sind.

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)

Ausgabe:

Screenshot eines Datenrahmens mit den Spalten