写一段lightgbm做回归的代码

最新推荐文章于 2024-07-02 01:16:37 发布

潮水岩

最新推荐文章于 2024-07-02 01:16:37 发布

阅读量204

点赞数

文章标签：回归机器学习算法 python 深度学习

本文链接：https://blog.csdn.net/weixin_42583683/article/details/129598858

版权

好的，下面是一段用 LightGBM 做回归的示例代码：

import lightgbmas lgb

# 读入数据
X_train, y_train = read_data()
X_test, y_test = read_data()

# 创建 LightGBM 训练数据集
lgb_train = lgb.Dataset(X_train, y_train)
lgb_eval = lgb.Dataset(X_test, y_test, reference=lgb_train)

# 定义参数
params = {
    'boosting_type': 'gbdt',