Python中的黄砖模块NotFoundError

Question

我试着用Yellowbrick做一个弯头图。(做k-means聚类)

我已经在jupyter笔记本上安装了Yellowbrick。但是，它总是像下面这样返回错误消息。

错误消息和信息附在下面的图片中。如果你能帮助我，我会很高兴的。

from yellowbrick.cluster import KElbowVisualizer

model = KMeans()

visualizer = KElbowVisualizer(model, k=(1,250))

visualizer.fit(x.reshape(-1,1))

ModuleNotFoundError                       Traceback (most recent call last)
<ipython-input-84-390153c57930> in <module>
----> 1 from yellowbrick.cluster import KElbowVisualizer
      2 model = KMeans()
      3 visualizer = KElbowVisualizer(model, k=(1,250))
      4 visualizer.fit(x.reshape(-1,1))
      5 

~/.local/lib/python3.7/site-packages/yellowbrick/__init__.py in <module>
     37 from .anscombe import anscombe
     38 from .datasaurus import datasaurus
---> 39 from .classifier import ROCAUC, ClassBalance, ClassificationScoreVisualizer
     40 
     41 # from .classifier import crplot, rocplot

~/.local/lib/python3.7/site-packages/yellowbrick/classifier/__init__.py in <module>
     24 from ..base import ScoreVisualizer
     25 from .base import ClassificationScoreVisualizer
---> 26 from .class_prediction_error import ClassPredictionError, class_prediction_error
     27 from .classification_report import ClassificationReport, classification_report
     28 from .confusion_matrix import ConfusionMatrix, confusion_matrix

~/.local/lib/python3.7/site-packages/yellowbrick/classifier/class_prediction_error.py in <module>
     22 
     23 from sklearn.utils.multiclass import unique_labels
---> 24 from sklearn.metrics._classification import _check_targets
     25 
     26 from yellowbrick.draw import bar_stack

ModuleNotFoundError: No module named 'sklearn.metrics._classification'

Answer 1

您好，感谢您查看Yellowbrick！

在sklearn v0.22中不推荐使用sklearn.metrics.classification模块，所以我们更新了我们的包，改为从sklearn.metrics._classification导入。

尝试更新您的scikit learn版本(例如pip install -U scikit-learn或conda update scikit-learn)，看看是否有帮助！

Answer 2

看起来你的黄砖没有正确安装。尝试仅为用户安装：

pip install -U yellowbrick

Answer 3

试试这个：

conda install -c districtdatalabs yellowbrick 

来源：https://anaconda.org/DistrictDataLabs/yellowbrick