从 scikit-learn 中的文件加载决策树答案

【问题标题】：load decision tree from file in scikit-learn从 scikit-learn 中的文件加载决策树
【发布时间】：2017-05-21 15:30:52
【问题描述】：

在 python scikit learn 中，有一个名为 export_graphviz 的方法可以将决策树导出到 dot 文件。

请问有没有办法将dot文件导入scikit learn作为决策树？像一些名为 sklearn.tree.import_graphviz() 的函数？

【问题讨论】：

那很难写。重新创建决策树所需的信息很可能不会被带入“点”文件并丢失。 “点”文件只是为了支持 graphviz，它是一个通用实用程序，并非专门设计用于表示决策树。我也没有在文档中看到必要的功能。
不可能。 graphviz 只是一张图片，表示您的树是如何以及在何处被分割的。

标签： python scikit-learn graphviz

【解决方案1】：

AFAIK 没有简单的方法可以做到这一点。 Graphviz 只能用于可视化决策树。如果您想保存模型，可以使用Pickle 保存模型。例如：

import cPickle
# save the classifier
with open('my_dumped_classifier.pkl', 'wb') as fid:
    cPickle.dump(gnb, fid)    

# load it again
with open('my_dumped_classifier.pkl', 'rb') as fid:
    gnb_loaded = cPickle.load(fid)

【讨论】：