class ui.module.MultilabelClassificationTask

Defines multi-label classification task.

Properties

MultilabelClassificationTask.num_classes

Number of classes.

int

MultilabelClassificationTask.class_names

Names of the classes.

list[str]

MultilabelClassificationTask.metric_to_monitor

Name of the default metric to monitor during validation.

str

MultilabelClassificationTask.metric_monitoring_mode

Whether to store values for maximum or minimum metric value.

MetricMonitoringMode

MultilabelClassificationTask.loss_function

The default loss function to use for training.

LossFunction

Methods

MulticlassClassificationTask.__init__(self, class_names)

from monad.ui.module import MultilabelClassificationTask

TARGET_NAMES = ["The North Face", "Adidas", "Tommy Hilfiger", "Hugo", "Lacoste", "Gap"]
task = MultilabelClassificationTask(class_names=TARGET_NAMES)

class_names: list[str]
The values of the target feature that will be used as class labels for multi-label classification.