DashAI.back.tasks.TextClassificationTask

class TextClassificationTask[source]

Base class for Text Classification Task.

Methods

`__init__`()
`get_metadata`()	Get metadata values for the current task
`num_labels`(dataset, output_column)	Get the number of unique labels in the output column.
`prepare_for_task`(datasetdict, outputs_columns)	Change the column types to suit the text classification task.
`process_predictions`(dataset, predictions, ...)	Process the predictions to return the class labels.
`validate_dataset_for_task`(dataset, ...)	Validate a dataset for the current task.

Attributes