YellowBrick 导入错误：无法从“sklearn.utils”导入名称“safe_indexing”

Question

我正在尝试为我运行的 K-Means 模型绘制轮廓图，但是，我收到错误：ImportError: cannot import name 'safe_indexing' from 'sklearn.utils.

我最初遇到了这篇文章中描述的 NotFoundError 问题，但是一旦我修复了它，我就遇到了这个问题。我的代码如下：

from yellowbrick.cluster import SilhouetteVisualizer

fig, ax = plt.subplots(2, 2, figsize=(15,8))
for i in range(1,6):
    
    kmeans = KMeans(n_clusters=i, init='k-means++', n_init=10, max_iter=100, random_state=0)
    q, mod = divmod(i, 2)
  
    visualizer = SilhouetteVisualizer(km, colors='yellowbrick', ax=ax[q-1][mod])
    visualizer.fit(ct1)

Answer 1

在带有yellowbrick v1.2 和scikit-learn v0.23.2 的python3.7 中，我没有收到错误，但是当我将scikit-learn 撞到v0.24.0 时，我得到了相同的ImportError。

我检查了 API 文档，就像在您引用的链接中一样，在 sklearn v0.22.0 中，它们已弃用utils.safe_indexing为私有 ( utils._safe_indexing)。但是我安装了yellowbrick v1.2 和v0.22.0<=sklearn<=v0.23.2 时没有出现导入错误。我认为他们最终utils.safe_indexing在 v0.24.0 中删除了 public ，所以如果你只是在 v0.24.0 之前安装一个 scikit-learn 版本，导入应该可以工作。

编辑：这是 Yellowbrick 的 github问题，如果您想跟踪他们的解决方法或更新此问题的进度