ai_classify

Klasifikuje obsah dokumentu do jednoho z poskytnutých popisků pomocí AI/LLM.

Odpovídající funkci SQL Databricks vizte funkceai_classify.

Syntax

from pyspark.sql import functions as dbf

dbf.ai_classify(col=<col>, labels=<labels>, options=<options>)

Parameters

Parametr	Typ	Description
col	pyspark.sql.Column nebo str	Sloupec obsahující obsah dokumentu, který se má klasifikovat.
labels	list, dict, pyspark.sql.Columnnebo str	Buď sada popisků literálů (Python seznam řetězců popisků nebo diktování mapující názvy popisků na popisy, serializované na literál JSON automaticky) nebo výraz sloupce, jehož hodnota na řádku je pole JSON řetězců popisků nebo objekt JSON mapující názvy popisků na popisy.
options	dictvolitelný	Slovník možností řízení chování klasifikace.

Returns

pyspark.sql.Column: Nový sloupec obsahující výsledek klasifikace.

Výchozí chování je klasifikace s jedním popiskem. Pokud chcete povolit klasifikaci více popisků a zobrazit úplnou sadu podporovaných možností, přečtěte si příručku jazyka SQL.

Příklady

# Static labels (same set for every row)
df.select(ai_classify("text", ["positive", "negative", "neutral"]))
df.select(ai_classify("text", {"positive": "Happy tone", "negative": "Unhappy tone"}))

# Per-row labels (a column whose value is a JSON array or JSON object string)
df.select(ai_classify("text", col("labels_json")))