一种回归树预测方法、控制装置及计算机可读存储介质
摘要:
本发明提供一种回归树预测方法、控制装置及计算机可读存储介质,方法包括如下步骤:输入训练数据集;用所述训练数据集训练回归树;判断训练得到的所述回归树的叶结点的数目是否大于3,若是,则采用詹姆斯-斯坦因估计同时估计所有所述叶结点的预测值;若否,则采用最大似然估计独立估计每个所述叶结点的预测值;输出所述回归树及所述预测值。在所有数据集上取得比基于最大似然估计的回归树更好的平均预测性能,即平均意义上更小的均方误差;在部分数据集上取得比现有最优的回归树预测方法--核回归更好的性能;在提升性能的同时,基本维持了原有基于最大似然估计预测的回归树的高效率,测试时间受数据集样本数量影响较小。
0/0