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



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

我想问一下是否有一种方法可以将点文件作为决策树导入scikit learn? 像某个叫做 sklearn.tree.import_graphviz() 的函数?

AFAIK 没有简单的方法可以做到这一点。图形可视化只能用于可视化决策树。如果要保存模型,可以使用 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)

相关内容

  • 没有找到相关文章

最新更新