如何导出以两个特征为矩阵的回归树?

时间:2016-09-16 13:52:23

标签: python r matrix tree regression

我有一个适合回归树应用的问题。最终结果需要在矩阵中,而不是树。在R或Python中有一种简单的方法可以导出一个回归树,它只有两个特征(在我的情况下是权重和距离)到一个矩阵,其中行是权重类,列是距离,单元格值是估计的价格?到目前为止,我只能找到树形可视化作为输出。

我认为,在每个维度上,树区分大约15个类,因此有200到300个单元格值,以防这很重要。

1 个答案:

答案 0 :(得分:0)

我真的不知道你的意思是“将树导出到矩阵”;我不知道回归树如何表示为矩阵。我想你想要显示你的(两个)特征/输入变量的多个组合的树的预测?

如果是这样,您首先需要创建所有感兴趣的组合,例如利用grid.expand。然后,将结果data.frame作为树的newdata方法的predict参数,以获得预测结果。最后,您将结果cast转换为矩阵形式。