Freigeben über


Verwenden Sie ai.embed mit Pandas

Die ai.embed Funktion verwendet generative KI, um Text in Vektoreinbettungen umzuwandeln. Mit diesen Vektoren können KI Beziehungen zwischen Texten verstehen, sodass Sie Inhalte basierend auf Bedeutung und nicht exakter Formulierung suchen, gruppieren und vergleichen können. Mit einer einzelnen Codezeile können Sie Vektoreinbettungen aus einer Spalte in einem DataFrame generieren.

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 Einbettungen als numpy-Array vom Typ float-32 für jede Eingabetextzeile enthält. Die Anzahl der Elemente im Array hängt von den Dimensionen des Einbettungsmodells 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)

Diese Beispielcodezelle stellt die folgende Ausgabe bereit:

Screenshot eines Datenrahmens mit den Spalten