Skip to content

utils_todo

infer_categorical_features

infer_categorical_features(
    X: ndarray, categorical_features
) -> List[int]

Infer the categorical features from the input data. We take self.categorical_features as the initial list of categorical features.

Parameters:

Name Type Description Default
X ndarray

The input data.

required

Returns:

Type Description
List[int]

Tuple[int, ...]: The indices of the categorical features.