Použite ai.extract s pandami

Funkcia ai.extract využíva generatívnu umelú inteligenciu na skenovanie vstupného textu a extrahovanie konkrétnych typov informácií označených štítkami, ktoré vyberiete (napríklad miesta alebo názvy). Používa iba jeden riadok kódu.

Poznámka

• Tento článok sa zaoberá používaním ai.extract s pandami. Ak chcete použiť ai.extract s PySparkom, pozrite si tento článok.
• Ďalšie funkcie umelej inteligencie nájdete v tomto prehľadnom článku.
• Naučte sa, ako prispôsobiť konfiguráciu funkcií AI.

Overview

Táto ai.extract funkcia rozširuje triedu série pandas . Ak chcete extrahovať vlastné typy entít z každého riadka vstupu, zavolajte funkciu v textovom stĺpci pandas DataFrame .

Na rozdiel od iných funkcií AI funkcia ai.extract vráti údajový rámec pandas namiesto radu so samostatným stĺpcom pre každý zadaný typ entity, ktorý obsahuje extrahované hodnoty pre každý vstupný riadok.

Syntax

df_entities = df["text"].ai.extract("entity1", "entity2", "entity3")

Parametre

Name	Popis
labels Povinné	Jeden alebo viac reťazcov , ktoré predstavujú množinu typov entít, ktoré sa majú extrahovať zo vstupných textových hodnôt.
aifunc.ExtractLabel Voliteľné	Jedna alebo viac definícií štítkov popisujúcich polia, ktoré treba extrahovať. Pre viac informácií pozri tabuľku parametrov ExtractLabel.

Parametre ExtractLabel

Name	Popis
label Povinné	Reťazec, ktorý reprezentuje entitu na extrakciu z hodnot vstupného textu.
description Voliteľné	Reťazec, ktorý pridáva ďalší kontext pre AI model. Môže obsahovať požiadavky, kontext alebo inštrukcie, ktoré má AI zvážiť pri extrakcii.
max_items Voliteľné	Int, ktorý určuje maximálny počet položiek na extrakciu pre tento štítok.
type Voliteľné	Typ JSON schémy pre extrahovanú hodnotu. Podporované typy pre túto triedu zahŕňajú string, number, integer, boolean, object, a .array
properties Voliteľné	Viac JSON schémových vlastností pre tento typ ako slovník. Môže obsahovať podporované vlastnosti ako "items" pre polia, "vlastnosti" pre objekty, "enum" pre enum typy a ďalšie. Pozri príklad použitia v tomto článku.
raw_col Voliteľné	Reťazec, ktorý nastavuje názov stĺpca pre surovú odpoveď LLM. Surová odpoveď poskytuje zoznam slovníkových párov pre každú entitnú značku, vrátane "reason" a "extraction_text".

Vrátené hodnoty

Funkcia vráti údajový rámec pandas so stĺpcom pre každý zadaný typ entity. Stĺpec alebo stĺpce obsahujú entity extrahované pre každý riadok vstupného textu. Ak funkcia identifikuje viac ako jednu zhodu pre entitu, vráti iba jednu z týchto zhôd. Ak sa zhoda nenájde, výsledok je null.

Predvolený typ návratu je zoznam reťazcov pre každý štítok. Ak používatelia zvolia v konfigurácii aifunc.ExtractLabel iný typ, napríklad "type=integer", výstupom bude zoznam python int. Ak používatelia v konfigurácii aifunc.ExtractLabel špecifikujú "max_items=1", pre daný štítok sa vráti iba jeden prvok typu.

Príklad

Iba vydavateľstvá

# This code uses AI. Always review output for mistakes.

df = pd.DataFrame([
        "MJ Lee lives in Tuscon, AZ, and works as a software engineer for Contoso.",
        "Kris Turner, a nurse at NYU Langone, is a resident of Jersey City, New Jersey."
    ], columns=["descriptions"])

df_entities = df["descriptions"].ai.extract("name", "profession", "city")
display(df_entities)

Táto príkladová bunka kódu poskytuje nasledujúci výstup:

ExtractLabel

# This code uses AI. Always review output for mistakes.

df = pd.DataFrame([
        "Alex Rivera, a 24-year-old midfielder from Barcelona, scored 12 goals last season, with an impressive 5 goals in one game.",
        "Jordan Smith, a 29-year-old striker from Manchester, scored exactly 1 goal in every game, for a total of 34 goals."
    ], columns=["bio"])

df["goals"] = df["bio"].ai.extract(
    aifunc.ExtractLabel(
        label = "goals", 
        description = "total goals only", 
        max_items = 1, 
        type = "integer"
    )
)
display(df)

Táto príkladová bunka kódu poskytuje nasledujúci výstup:

Súvisiaci obsah

• Použite ai.extract s PySparkom.

• Zistite sentiment pomocou ai.analyze_sentiment.

• Kategorizujte text pomocou ai.classify.

• Generujte vektorové embeddingy pomocou ai.embed.

• Opravte gramatiku pomocou ai.fix_grammar.

• Odpovedzte na vlastné výzvy používateľov pomocou ai.generate_response.

• Vypočítajte podobnosť s ai.podobnosťou.

• Zhrňte text pomocou ai.summarize.

• Preložte text pomocou ai.translate.

• Prečítajte si viac o úplnej sade funkcií umelej inteligencie.

• Prispôsobte konfiguráciu funkcií AI.

• Chýbala vám funkcia, ktorú potrebujete? Navrhnite ho na fóre Fabric Ideas.