get_columns_by_role

lightautoml.pipelines.utils.get_columns_by_role(dataset, role_name, **kwargs)[source]

Search for columns with specific role and attributes when building pipeline.

Parameters:

  • dataset (LAMLDataset) – Dataset to search.

  • role_name (str) – Name of features role.

  • **kwargs (Any) – Specific parameters values to search. Example: search for categories with OHE processing only.

Return type:

List[str]

Returns:

List of str features names.