inspection.importance.plot_feature_importances
inspection.importance.plot_feature_importances(
X,
y,
feature_names,
target_names,
target_index,
n_top_features=10,
figsize=(6, 6),
)
Generate and plot feature importances using MDI and permutation importance.
Parameters
| X |
np.ndarray |
Input features array |
required |
| y |
np.ndarray |
Target array |
required |
| feature_names |
list |
List of feature names |
required |
| target_names |
list |
List of target names |
required |
| target_index |
int |
Index of target variable to analyze |
required |
| n_top_features |
int |
Number of top features to show |
10 |
| figsize |
tuple |
Size of the figure |
(6, 6) |
Returns
| tuple |
tuple |
(top_features, importance_df) |
Examples
>>> from spotoptim.sensitivity import plot_feature_importances
>>> import pandas as pd
>>> from sklearn.datasets import make_regression
>>> X, y = make_regression(n_samples=100, n_features=5, noise=0.1, random_state=42)
>>> feature_names = [f"feature_{i}" for i in range(X.shape[1])]
>>> target_names = ["target"]
>>> top_features, imp_df = plot_feature_importances(X, y, feature_names, target_names, target_index=0)
>>> print("Top features:", top_features)