Class EvaluationTextClassificationSpec (1.48.0)

EvaluationTextClassificationSpec(
    ground_truth_data: typing.Union[typing.List[str], str, pandas.DataFrame],
    target_column_name: str,
    class_names: typing.List[str],
)

Spec for text classification model evaluation tasks.

Parameters

NameDescription
target_column_name str

Required. The label column in the dataset provided in ground_truth_data. Required when task_name='text-classification'.

class_names List[str]

Required. A list of all possible label names in your dataset. Required when task_name='text-classification'.