random-forest-importances icon indicating copy to clipboard operation
random-forest-importances copied to clipboard

Support for alternative metrics for `oob_dropcol_importances`

Open ValterH opened this issue 1 year ago • 0 comments

The default implementation of oob_dropcol_importances sets the oob_score attribute to true. This results the method always returning the default ($R^2$) score. Adding a keyword argument for metric of interest would greatly improve the method's flexibility. I would be happy to submit a PR as the implementation would be quite simple.

Desired outcome: drop_imp = oob_dropcol_importances(model, X_boot, y_boot, oob_metric=negative_rmse)

ValterH avatar Sep 26 '24 13:09 ValterH