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