NumericRole

class lightautoml.dataset.roles.NumericRole(dtype=numpy.float32, force_input=False, prob=False, discretization=False)

Bases: lightautoml.dataset.roles.ColumnRole

Numeric role.

__init__(dtype=numpy.float32, force_input=False, prob=False, discretization=False)

Create numeric role with specific numeric dtype.

Parameters

• dtype (Union[Callable, str]) – Variable type.

• force_input (bool) – Select a feature for training, regardless of the selector results.

• prob (bool) – If input number is probability.