1

我正在尝试使用 Yellowbrick 制作肘部图。（制作 k-means 聚类）

我已经在 jupyter notebook 中安装了 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'
4

3 回答 3

2

您好，感谢您查看 Yellowbrick！

该sklearn.metrics.classification模块在 sklearn v0.22 中已被弃用，因此我们更新了我们的包以sklearn.metrics._classification代替导入。

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

于 2020-11-24T00:26:30.310 回答
1

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

pip install -U yellowbrick
于 2020-11-23T06:59:43.277 回答
0

尝试这个：

conda install -c districtdatalabs yellowbrick

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

于 2021-09-26T13:11:53.810 回答