You are viewing documentation about an older version (1.44.0). View latest version

snowflake.snowpark.functions.ai_summarize_agg¶

snowflake.snowpark.functions.ai_summarize_agg(expr: Union[Column, str]) → Column[source]¶

Summarizes a column of text data.

Parameters:: expr – This is an expression that contains text for summarization, such as restaurant reviews or phone transcripts.

Example:

>>> df = session.create_dataframe([
...     [1, "Excellent"],
...     [1, "Excellent"],
...     [1, "Great"],
...     [1, "Mediocre"],
...     [2, "Terrible"],
...     [2, "Bad"],
... ], schema=["product_id", "review"])
>>> summary_df = df.select(ai_summarize_agg(col("review")))
>>> summary_df.count()
1
>>> summary_df = df.group_by("product_id").agg(ai_summarize_agg(col("review")))
>>> summary_df.count()
2