2013-12-12 95 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

相關問題