我正在使用下一个代码来训练一个带有一个节点(Decision stump)的ClassificationTree,使用CutPoint属性我能够获得谓词拆分的值,但是如何获得使用的谓词? (在此示例中为feature1或feature2)
x1 = 10 * rand(10, 1);
x2 = 10 * rand(10, 1);
x = [x1 x2];
y = randi(2, 10, 1);
t = ClassificationTree.fit(x, y, 'MinParent', 10, 'MergeLeaves', 'off');
view(t)
t.CutPoint(1,:)
输出
Decision tree for classification
1 if x1<3.60576 then node 2 elseif x1>=3.60576 then node 3 else 1
2 class = 1
3 class = 2
ans =
3.6058
所以我能够得到值3.6058,但我需要一种方法来获得x1是使用的功能。
答案 0 :(得分:1)
我想你想要:
t.CutVar{1}
ans =
x1