matchzoo.tasks.classification
¶
Classification task.
Module Contents¶
Classes¶
Classification task. |
-
class
matchzoo.tasks.classification.
Classification
(num_classes: int = 2, **kwargs)¶ Bases:
matchzoo.engine.base_task.BaseTask
Classification task.
Examples
>>> classification_task = Classification(num_classes=2) >>> classification_task.metrics = ['acc'] >>> classification_task.num_classes 2 >>> classification_task.output_shape (2,) >>> classification_task.output_dtype <class 'int'> >>> print(classification_task) Classification Task with 2 classes
-
TYPE
= classification¶
-
property
num_classes
(self) → int¶ - Returns
number of classes to classify.
-
classmethod
list_available_losses
(cls) → list¶ - Returns
a list of available losses.
-
classmethod
list_available_metrics
(cls) → list¶ - Returns
a list of available metrics.
-
property
output_shape
(self) → tuple¶ - Returns
output shape of a single sample of the task.
-
property
output_dtype
(self)¶ - Returns
target data type, expect int as output.
-
__str__
(self)¶ - Returns
Task name as string.
-