2015-05-10 29 views
0

我使用熊猫指令tree.DecisionTreeClassifier来构建(二元)分类树。沿着线的东西:得到熊猫的实际决策树

dcrG = tree.DecisionTreeClassifier(criterion='entropy',splitter='best',options_go_here) 
dcrG.fit(train[features], train['G']) 

现在,我已经成功地建立了我的决策树,我想熊猫打印我出实际决策树,这样的东西沿着

if (var1>0.4) 
    if (var4>3.24) 
    if (var2<0.5) 
     return 1 
    else 
     return 0 
    else 
    return 1 
else 
    if (var3>3.5) 
    if (var2<0.1) 
     return 0 
    else 
     return 1 
    else 
    if (var2>0.4) 
     return 1 
    else 
     return 0 

等都以线条我可以将得到的算法导出到其他编程语言。我怎样才能做到这一点?

+1

见:http://stackoverflow.com/questions/20224526/how-to-extract-the-decision-rules-from-scikit-learn-决策树 – user308827

+0

阿哈,显然在我的搜索中使用了错误的关键字。谢谢! – user1111929

回答