matchzoo.tasks.classification

Classification task.

Module Contents

Classes

Classification

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.