DataType

class torch_brain.registry.DataType(value)

Bases: enum.Enum

Enum defining the possible data types.

CONTINUOUS

For continuous-valued variables

BINARY

For binary variables

MULTINOMIAL

For multi-class variables

MULTILABEL

For multi-label variables