8000 EXA add compatibility with scipy 1.5 (#17695) · scikit-learn/scikit-learn@0fb307b · GitHub
[go: up one dir, main page]

Skip to content

Commit 0fb307b

Browse files
committed
EXA add compatibility with scipy 1.5 (#17695)
1 parent 0186bf3 commit 0fb307b

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

examples/inspection/plot_permutation_importance_multicollinear.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -80,8 +80,9 @@
8080
fig, (ax1, ax2) = plt.subplots(1, 2, figsize=(12, 8))
8181
corr = spearmanr(X).correlation
8282
corr_linkage = hierarchy.ward(corr)
83-
dendro = hierarchy.dendrogram(corr_linkage, labels=data.feature_names, ax=ax1,
84-
leaf_rotation=90)
83+
dendro = hierarchy.dendrogram(
84+
corr_linkage, labels=data.feature_names.tolist(), ax=ax1, leaf_rotation=90
85+
)
8586
dendro_idx = np.arange(0, len(dendro['ivl']))
8687

8788
ax2.imshow(corr[dendro['leaves'], :][:, dendro['leaves']])

0 commit comments

Comments
 (0)
0