導入錯誤：無法從“sklearn.tree”導入名稱“plot_tree”

Question

從 sklearn 導入 plot_tree 時導入錯誤

做pip install sklearn 后

錯誤日志：

from sklearn.tree import DecisionTreeClassifier, plot_tree

ImportError: cannot import name 'plot_tree' from 'sklearn.tree'

Answer 1

sklearn 版本 > 0.21 后可以使用繪圖樹。

pip install --upgrade sklearn

可能會有所幫助，但如果不是，則必須升級整個 python 版本。

我在 3.6.7 python 上遇到了同樣的問題，並通過安裝 3.7.7 來解決它。

Answer 2

安裝 scikit-learn

使固定：

pip install scikit-learn