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

Posted 2023-03-12

【中文标题】从 scikit-learn 中的文件加载决策树

【英文标题】：load decision tree from file in scikit-learn

【发布时间】：2017-05-21 15:30:52

【问题描述】：

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

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

【问题讨论】：

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

不可能。 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)