Python:从AdaBoostClassifier获取规则

时间:2015-03-10 10:55:09

标签: python adaboost

我在Python中使用AdaBoostClassifier(来自sklearn.ensemble导入AdaBoostClassifier),我想知道AdaBoost选择的弱规则。

这是我的源代码:

x = np.array(p_values_learn) #Array of 10.000 * 100.000 values are float betweek 0 and 1
y = np.array(verite_learn) #Vector of 100.000 values are 0 or 1
bdt = AdaBoostClassifier(algorithm="SAMME.R", n_estimators=4)
bdt.fit(x, y)

每个估算器都是DecisionTreeClassifier,但我无法找到我想要的信息。

我想知道决定函数f(x)的规则细节:

  

f(x)= 0.426 I(x37 <2.5)+ 0.64696 I(x250 <8.5)

那就是说,我想知道分类器使用哪个数据X列以及哪个系数。

它是一个二元决策,类是0或1。

感谢。

1 个答案:

答案 0 :(得分:0)

根据描述,我认为decision_function(X)正是您所寻找的: decision_function