r - 需要包树分类结果的其他输出

Question

具有多个类值的类目标字段的包的树分类输出tree如下所示：

      30-39     40-49     50-59     60plus       to29
1 0.1723497 0.2145071 0.5151891 0.02588345 0.07207068
2 0.1451786 0.1370050 0.2740099 0.32061886 0.12318770
4 0.1723497 0.2145071 0.5151891 0.02588345 0.07207068

但我希望结果是这样的：

1  50-59
2  60plus
4  50-59

是否有任何已实现的函数（我还没有找到）可以自动提供此输出，还是我必须编写一个 R 函数？

该函数必须获取每行的最大值，获取该行中匹配值的索引，然后提供该索引的年龄类别，如60plus.

Answer 1

在predict函数中有一个名为 的参数type。它定义了预测所需的输出类型。

做就是了

predict(treeobject, type="class")

你会得到更高概率的类别。

有关详细信息，请参阅参考手册。(predict.tree 函数)