从随机森林分类器中提取树(预测器)

Question

我有一个关于sklearn,随机森林分类器的具体技术问题.

在使用".fit(X,y)"方法拟合数据之后,有没有办法以某种通用格式从估计器对象中提取实际树,因此可以在外部实现".predict(X)"方法蟒蛇？

Answer 1

是的,林的树存储在estimators_林对象的属性中.

您可以查看export_graphviz 函数的实现,以了解如何编写自定义导出器:

https://github.com/scikit-learn/scikit-learn/blob/master/sklearn/tree/export.py

以下是此功能的使用文档:

http://scikit-learn.org/stable/modules/tree.html#classification