features#

Utils module containing feature importance calculations.

Functions

calculate_feature_importance_or_none(model, ...)

Calculate features effect on the label or None if the input is incorrect.

column_importance_sorter_dict(cols_dict, dataset)

Return the dict of columns sorted and limited by feature importance.

column_importance_sorter_df(df, ds, ...[, ...])

Return the dataframe of columns sorted and limited by feature importance.

infer_categorical_features(df[, ...])

Infers which features are categorical by checking types and number of unique values.

infer_numerical_features(df)

Infers which features are numerical.

is_categorical(column[, ...])

Check if uniques are few enough to count as categorical.