2013-12-12 56 views
7

我有一个关于sklearn,随机森林分类器的特定技术问题。从随机森林分类器中提取树木(预测器)

数据与“.fit(X,Y)”法进行拟合后, 是有办法来提取来自估计对象的实际树木 ,在一些通用格式,因此,” .predict(X) “ 方法可以在python之外实现吗?

回答

12

是的,森林的树木被存储在森林物体的estimators_属性 中。

你可以看看export_graphviz 功能的实现学习了编写自定义导出:

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

以下是此功能的使用DOC:

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

+0

谢谢Olivier!这真的有帮助 – user3095701

+1

有没有办法将其解压缩为JSON格式? – user3095701

+0

还没有(截至0.14.1)。这里有一个拉请求:https://github.com/scikit-learn/scikit-learn/pull/2530 – ogrisel