Del via


Bruk ai.embed med pandas

Funksjonen ai.embed bruker generativ AI for å konvertere tekst til vektorinnlegginger. Disse vektorene lar AI forstå sammenhenger mellom tekster, slik at du kan søke, gruppere og sammenligne innhold basert på mening i stedet for eksakt formulering. Med én enkelt kodelinje kan du generere vektorinnlegginger fra en kolonne i en DataFrame.

Note

Oversikt

Funksjonen ai.embedutvider pandas Series-klassen .

For å generere vektorinnleggelser av hver inngangsrad, kall funksjonen enten på en pandas-serie eller en tekstkolonne i pandas DataFrame.

Funksjonen returnerer en pandas-serie som inneholder embeddinger, som kan lagres i en ny DataFrame-kolonne.

Syntaks

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

Parametere

Ingen.

Returnerer

Funksjonen returnerer en pandas-serie som inneholder embeddinger som numpy array av float-32 for hver inndatatekstrad. Antallet elementer i arrayet avhenger av embeddingmodellens dimensjoner, som kan konfigureres i AI-funksjoner

Eksempel

# 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)

Denne eksempelkodecellen gir følgende utdata:

Skjermbilde av en dataramme med kolonnene 'beskrivelser' og 'embed'. Kolonnen 'embed' inneholder embeddings for beskrivelsene.