Använd ai.embed med pandas

Funktionen ai.embed använder generativ AI för att konvertera text till vektorbäddningar. Med dessa vektorer kan AI förstå relationer mellan texter, så att du kan söka, gruppera och jämföra innehåll baserat på innebörd snarare än exakta formuleringar. Med en enda kodrad kan du generera vektorbäddningar från en kolumn i en DataFrame.

Anmärkning

• Den här artikeln beskriver hur du använder ai.embed med Pandas. Information om hur du använder ai.embed med PySpark finns i den här artikeln.
• Se andra AI-funktioner i den här översiktsartikeln.
• Lär dig hur du anpassar konfigurationen av AI-funktioner.

Översikt

Klassen ai.embed utökas av funktionen .

Om du vill generera vektorinbäddningar för varje indatarad anropar du funktionen på antingen en Pandas-serie eller en textkolumn i Pandas DataFrame.

Funktionen returnerar en Pandas-serie som innehåller inbäddningar, som kan lagras i en ny DataFrame-kolumn.

Syntax

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

Parameterar

Ingen.

Retur

Funktionen returnerar en Pandas-serie som innehåller inbäddningar som numpy-matris med float-32 för varje indatatextrad. Antalet element i matrisen beror på inbäddningsmodellens dimensioner, som kan konfigureras i AI-funktioner

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)

Den här exempelkodcellen innehåller följande utdata:

Relaterat innehåll

• Använd ai.embed med PySpark.

• Identifiera sentiment med ai.analyze_sentiment.

• Kategorisera text med ai.classify.

• Extrahera entiteter med ai_extract.

• Åtgärda grammatik med ai.fix_grammar.

• Svara på anpassade användarfrågor med ai.generate_response.

• Beräkna likhet med ai.similarity.

• Sammanfatta text med ai.summarize.

• Översätt text med ai.translate.

• Läs mer om den fullständiga uppsättningen AI-funktioner.

• Anpassa konfigurationen av AI-funktioner.

• Missade vi en funktion som du behöver? Föreslå det på forumet Fabric Ideas.