我尝试从基于单个变量创建的决策树中获取拆分。下面的方法是正确和安全的吗?
from sklearn.tree import DecisionTreeClassifier
import numpy as np
x = np.array([1,2,3,4,5,6,7])
y = np.array([1,0,0,1,1,0,1])
x = x.reshape(7, -1)
clf = DecisionTreeClassifier()
clf.fit(x, y)
# My splits
np.sort(clf.tree_.threshold[clf.tree_.feature == 0])
我只看到
0
和
-2
在里面
clf.tree_.fature
我的理解是
TREE_UNDEFINED
here
)而其他点将是具有定义的阈值的leafs。