roles_parser

lightautoml.dataset.utils.roles_parser(init_roles)

Parser of roles.

Parse roles from old format numeric: [var1, var2, ...] to {var1:numeric, var2:numeric, ...}.

Parameters:

init_roles (Dict[Union[ColumnRole, str], Union[str, Sequence[str]]]) – Mapping between roles and feature names.

Return type:

Dict[str, ColumnRole]

Returns:

Roles dict in format key - feature names, value - roles.