【发布时间】:2019-09-17 14:03:04
【问题描述】:
我正在写一个决策树,下面的代码是完整代码的一部分:
def show_tree(tree, features, path):
f = io.StringIO()
export_graphviz(tree, out_file=f, feature_names=features)
pydotplus.graph_from_dot_data(f.getvalue()).write_png(path)
img = misc.imread(path)
plt.rcParams['figure.figsize'] = (20,20)
plt.imshow(img)
谁能告诉我在这里使用StringIO的目的是什么?
【问题讨论】:
标签: python scikit-learn decision-tree stringio