API Reference

MulticlassClassificationTask

class ui.module.MulticlassClassificationTask

Defines multiclass classification task.

Properties

MulticlassClassificationTask.num_classes

Number of classes

Returns

int

MulticlassClassificationTask.class_names

Names of the classes

Returns

list[str]

MulticlassClassificationTask.metric_to_monitor

Name of the default metric to monitor during validation.

Returns

str

MulticlassClassificationTask.metric_monitoring_mode

Whether to store values for maximum or minimum metric value.

Returns

MetricMonitoringMode

MulticlassClassificationTask.loss_function

The default loss function to use for training.

Returns

LossFunction

Methods

MulticlassClassificationTask.__init__(self, class_names)

from monad.ui.module import MulticlassClassificationTask

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

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