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

Question

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

我想询问是否有一种方法可以将点文件导入scikit学习作为决策树？像一些名为sklearn.tree.import_graphviz（）的函数？

Answer 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)