梯度提升回归树 GBDT java

最新推荐文章于 2024-08-16 22:44:00 发布

九佾

最新推荐文章于 2024-08-16 22:44:00 发布

阅读量3.1k

点赞数

本文链接：https://blog.csdn.net/ysh126/article/details/53119194

版权

该博客介绍了一个简单的Java实现的梯度提升回归树（GBDT）算法。通过加载数据、更新训练集目标和迭代训练，展示了如何使用GBDT进行回归预测。在每个迭代过程中，创建并训练决策树，然后更新目标值。最后，使用测试数据评估模型性能。

摘要由CSDN通过智能技术生成

/**
* 梯度提升回归树    简单实现
* @author ysh 1208706282
*
*/
public class Gbdt {
   static List<Sample> mSamples;
   static List<Double> mTrainTarget;
   static List<Double> mTrainCurTarget;
   static List<Cart> mCarts;
   /**
   * 加载数据   回归树
   * @param path
   * @param regex
   * @throws Exception
   */
   public void loadData(String path,String regex) throws Exception{
       mSamples = new ArrayList<Sample>();
       mTrainTarget = new ArrayList<Double>();
       mTrainCurTarget = new ArrayList<Double>();
       BufferedReader reader = new BufferedReader(new FileReader(path));
       String line = null;
       String splits[] = null;
       Sample sample = null;
       while(null != (