为了解决export_graphviz画决策树图时中文乱码的问题,需要使用sklearn.externals.six包里的StringIO,显示如下报错:
ModuleNotFoundError: No module named 'sklearn.externals.six'
查阅资料后发现sklearn 0.23版本已经删掉了这个包。
因此直接安装six包:
pip install six
然后直接引入即可:
from six import StringIO
问题解决