执行代码:
from six import StringIO
from IPython.display import Image
import pydot
import pydotplus
dot_data = StringIO()
tree.export_graphviz(credit_model, out_file = dot_data,
feature_names = X_train.columns,
class_names=['no default','default'],
filled = True, rounded = True,
special_characters = True)
graph = pydot.graph_from_dot_data(dot_data.getvalue())
Image(graph.create_png())
报错信息:
AttributeError Traceback (most recent call last) <ipython-input-19-467e13633827> in <module> 10 special_characters = True) 11 graph = pydot.graph_from_dot_data(dot_data.getvalue()) ---> 12 Image(graph.create_png()) AttributeError: 'list' object has no attribute 'create_png'
解决办法:
更改语句为:graph, = pydot.graph_from_dot_data(dot_data.getvalue())
执行成功