search

JuliaAI / MLJTuning.jl

Hyperparameter optimization algorithms for use in the MLJ machine learning framework
MIT License
66 stars 12 forks source link

add feature importances support for tuned models #213

Closed OkonSamuel closed 3 months ago

OkonSamuel commented 4 months ago

This PR adds MLJModelInterface.feature_importances support for TunedModel.

Now calling MLJModelInterface.feature_importances on a TunedModel object just returns the feature importances for the wrapped model. For cases where the model wrapped by the TunedModel object doesn't support feature importance, nothing is returned.

codecov[bot] commented 4 months ago

Codecov Report

All modified and coverable lines are covered by tests :white_check_mark:

Project coverage is 87.53%. Comparing base (c081bd0) to head (b9d8c01). Report is 4 commits behind head on dev.

Additional details and impacted files ```diff @@ Coverage Diff @@ ## dev #213 +/- ## ========================================== + Coverage 87.42% 87.53% +0.11% ========================================== Files 13 13 Lines 660 666 +6 ========================================== + Hits 577 583 +6 Misses 83 83 ```

:umbrella: View full report in Codecov by Sentry.
:loudspeaker: Have feedback on the report? Share it here.