获取用于在matlab ClassificationTree中拆分值的谓词

时间:2013-08-20 11:23:59

标签: matlab classification decision-tree

我正在使用下一个代码来训练一个带有一个节点(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是使用的功能。

1 个答案:

答案 0 :(得分:1)

我想你想要:

t.CutVar{1}

ans =
x1