Skip to content

AI Classify

AI Classify is a transformation component that uses the Databricks ai_classify() function to invoke generative AI to classify input text according to labels you provide. This function uses a Databricks chat model serving endpoint made available by Databricks Foundation Model APIs.

The output is a new column named classification_<input-column-name>, which contains the most appropriate label from the provided list.


Make sure you have read and understand the Requirements set out by Databricks before using this component.


Name = string

A human-readable name for the component.

Column = drop-down

Select an input column that contains the text to be classified.


The component will operate on only a single column of the input.

Classification Labels = column editor

Enter a list of labels which will be used to classify the input text. Enter at least two labels, and no more than 20.

Enter one label per row in the Classification Labels dialog. Click + to add a new row.

Include Input Columns = boolean

  • Yes: Outputs both your source input columns and the classification label column. This will also include those input columns not selected in Column.
  • No: Only includes the classification label columns.

Snowflake Databricks Amazon Redshift