python - 导入错误 : No module name sklearn. external.six

Question

我目前正在使用 anaconda 4.8.3 并想显示决策 TreeMap ，我在 anaconda 中安装了 graphviz 和 pydotplus 库而不是这个我收到错误“ModuleNotFoundError: No module named 'sklearn.externals.six' .这是我的代码:

from sklearn.tree import DecisionTreeClassifier
from IPython.display import Image  
from sklearn.externals.six import StringIO  
from sklearn.tree import export_graphviz
import pydot 
features = list(df.columns[1:])
features

这是我的错误:

ModuleNotFoundError   Traceback (most recent call last)
<ipython-input-19-0b3416ce7fda> in <module>

        1 from IPython.display import Image
--->    2 from sklearn.externals.six import StringIO
        3 from sklearn.tree import export_graphviz
        4 import pydot
        5 ModuleNotFoundError: No module named 'sklearn.externals.six'

Answer 1

您可以直接从模块六导入StringIO，无需降级scikit。

from six import StringIO