mlproject.postprocess.feature_importances

mlproject.postprocess.feature_importances#

Functions for getting feature importances data from ARFS, PFI and SHAP values from trained models

Functions

aggregate_importances

Aggregate feature importance DataFrames across folds.

get_arfs_mean_feature_importances

Get mean feature importances across GrootCV runs.

get_modnet_pfi_shap_summary

Load trained MODNet models (5 folds), compute PFI + SHAP values

get_rf_pfi_shap_summary

Load trained RandomForest models (5 folds), compute PFI + SHAP values

process_dataset

Process MODData to get scaled feature array for a given MODNetModel.