EXA remove warning when setting yticks in permutation importances exa…

…mples (scikit-learn#19385)
j3nnn1 · Feb 7, 2021 · 9433799 · 9433799
1 parent bd67f12
commit 9433799
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/examples/inspection/plot_permutation_importance.py b/examples/inspection/plot_permutation_importance.py
@@ -130,8 +130,8 @@
 y_ticks = np.arange(0, len(feature_names))
 fig, ax = plt.subplots()
 ax.barh(y_ticks, tree_feature_importances[sorted_idx])
-ax.set_yticklabels(feature_names[sorted_idx])
 ax.set_yticks(y_ticks)
+ax.set_yticklabels(feature_names[sorted_idx])
 ax.set_title("Random Forest Feature Importances (MDI)")
 fig.tight_layout()
 plt.show()

diff --git a/examples/inspection/plot_permutation_importance_multicollinear.py b/examples/inspection/plot_permutation_importance_multicollinear.py
@@ -60,8 +60,8 @@
 fig, (ax1, ax2) = plt.subplots(1, 2, figsize=(12, 8))
 ax1.barh(tree_indices,
          clf.feature_importances_[tree_importance_sorted_idx], height=0.7)
-ax1.set_yticklabels(data.feature_names[tree_importance_sorted_idx])
 ax1.set_yticks(tree_indices)
+ax1.set_yticklabels(data.feature_names[tree_importance_sorted_idx])
 ax1.set_ylim((0, len(clf.feature_importances_)))
 ax2.boxplot(result.importances[perm_sorted_idx].T, vert=False,
             labels=data.feature_names[perm_sorted_idx])