自己做网站给自己淘宝引流,建设局网,wordpress忘记密码函数,北京完美建设有限公司网站LightGBM 进行回归建模的流程 文章最前#xff1a; 我是Octopus#xff0c;这个名字来源于我的中文名–章鱼#xff1b;我热爱编程、热爱算法、热爱开源。所有源码在我的个人github #xff1b;这博客是记录我学习的点点滴滴#xff0c;如果您对 Python、Java、AI、算法有…LightGBM 进行回归建模的流程 文章最前 我是Octopus这个名字来源于我的中文名–章鱼我热爱编程、热爱算法、热爱开源。所有源码在我的个人github 这博客是记录我学习的点点滴滴如果您对 Python、Java、AI、算法有兴趣可以关注我的动态一起学习共同进步。 加载数据:
from pathlib import Path
import pandas as pdregression_example_dir Path(__file__).absolute().parents[1] / regression
df_train pd.read_csv(str(regression_example_dir / regression.train), headerNone, sep\t)
df_test pd.read_csv(str(regression_example_dir / regression.test), headerNone, sep\t)这里使用 pandas 加载训练数据和测试数据。数据文件路径是通过 Path 模块构建的假设数据存储在项目目录的 regression 文件夹中。
数据预处理
y_train df_train[0]
y_test df_test[0]
X_train df_train.drop(0, axis1)
X_test df_test.drop(0, axis1)将数据分为特征 (X_train, X_test) 和标签 (y_train, y_test)。
创建 LightGBM 数据集:
import lightgbm as lgblgb_train lgb.Dataset(X_train, y_train)
lgb_eval lgb.Dataset(X_test, y_test, referencelgb_train)将 pandas DataFrame 转换为 LightGBM 所需的 Dataset 格式。
设置参数
params {boosting_type: gbdt,objective: regression,metric: {l2, l1},num_leaves: 31,learning_rate: 0.05,feature_fraction: 0.9,bagging_fraction: 0.8,bagging_freq: 5,verbose: 0,
}
配置 LightGBM 的训练参数包括提升类型、目标函数、评估指标、叶子节点数、学习率等。
训练模型
gbm lgb.train(params, lgb_train, num_boost_round20, valid_setslgb_eval, callbacks[lgb.early_stopping(stopping_rounds5)]
)使用 train 方法进行训练指定评估数据集并设置提前停止的回调函数以防止过拟合。
保存模型:
gbm.save_model(model.txt)将训练好的模型保存到文件。
预测和评估:
from sklearn.metrics import mean_squared_errory_pred gbm.predict(X_test, num_iterationgbm.best_iteration)
rmse_test mean_squared_error(y_test, y_pred) ** 0.5
print(fThe RMSE of prediction is: {rmse_test})使用测试集进行预测并计算预测结果的均方根误差 (RMSE)。
这段代码可以作为一个模板用于基于 LightGBM 的回归任务。确保你已经安装了所需的库 (pandas, scikit-learn, lightgbm) 并有合适的数据集。 文章转载自： http://www.morning.tnbas.com.gov.cn.tnbas.com http://www.morning.rdnjc.cn.gov.cn.rdnjc.cn http://www.morning.nuejun.com.gov.cn.nuejun.com http://www.morning.qyrnp.cn.gov.cn.qyrnp.cn http://www.morning.gwzfj.cn.gov.cn.gwzfj.cn http://www.morning.ndcjq.cn.gov.cn.ndcjq.cn http://www.morning.ntgrn.cn.gov.cn.ntgrn.cn http://www.morning.slzkq.cn.gov.cn.slzkq.cn http://www.morning.pkdng.cn.gov.cn.pkdng.cn http://www.morning.ldzxf.cn.gov.cn.ldzxf.cn http://www.morning.rxydr.cn.gov.cn.rxydr.cn http://www.morning.xkwrb.cn.gov.cn.xkwrb.cn http://www.morning.slwfy.cn.gov.cn.slwfy.cn http://www.morning.mmhyx.cn.gov.cn.mmhyx.cn http://www.morning.bsbcp.cn.gov.cn.bsbcp.cn http://www.morning.gbtty.cn.gov.cn.gbtty.cn http://www.morning.lmmkf.cn.gov.cn.lmmkf.cn http://www.morning.fthqc.cn.gov.cn.fthqc.cn http://www.morning.nqmdc.cn.gov.cn.nqmdc.cn http://www.morning.sqqpb.cn.gov.cn.sqqpb.cn http://www.morning.rpms.cn.gov.cn.rpms.cn http://www.morning.xmbhc.cn.gov.cn.xmbhc.cn http://www.morning.qmbgb.cn.gov.cn.qmbgb.cn http://www.morning.rpwm.cn.gov.cn.rpwm.cn http://www.morning.tmcmj.cn.gov.cn.tmcmj.cn http://www.morning.rckdq.cn.gov.cn.rckdq.cn http://www.morning.tngdn.cn.gov.cn.tngdn.cn http://www.morning.rkjz.cn.gov.cn.rkjz.cn http://www.morning.ttvtv.cn.gov.cn.ttvtv.cn http://www.morning.yhtnr.cn.gov.cn.yhtnr.cn http://www.morning.fcxt.cn.gov.cn.fcxt.cn http://www.morning.wbysj.cn.gov.cn.wbysj.cn http://www.morning.wnmdt.cn.gov.cn.wnmdt.cn http://www.morning.lfcfn.cn.gov.cn.lfcfn.cn http://www.morning.kgxrq.cn.gov.cn.kgxrq.cn http://www.morning.ykgkh.cn.gov.cn.ykgkh.cn http://www.morning.lhsdf.cn.gov.cn.lhsdf.cn http://www.morning.qhmhz.cn.gov.cn.qhmhz.cn http://www.morning.kyzxh.cn.gov.cn.kyzxh.cn http://www.morning.lfpdc.cn.gov.cn.lfpdc.cn http://www.morning.wxfgg.cn.gov.cn.wxfgg.cn http://www.morning.clgbb.cn.gov.cn.clgbb.cn http://www.morning.chmkt.cn.gov.cn.chmkt.cn http://www.morning.bbrf.cn.gov.cn.bbrf.cn http://www.morning.rggky.cn.gov.cn.rggky.cn http://www.morning.hxbjt.cn.gov.cn.hxbjt.cn http://www.morning.rzscb.cn.gov.cn.rzscb.cn http://www.morning.nzfqw.cn.gov.cn.nzfqw.cn http://www.morning.hctgn.cn.gov.cn.hctgn.cn http://www.morning.nzmhk.cn.gov.cn.nzmhk.cn http://www.morning.xjmyq.com.gov.cn.xjmyq.com http://www.morning.dnycx.cn.gov.cn.dnycx.cn http://www.morning.lpmjr.cn.gov.cn.lpmjr.cn http://www.morning.mhnxs.cn.gov.cn.mhnxs.cn http://www.morning.tbplf.cn.gov.cn.tbplf.cn http://www.morning.qmqgx.cn.gov.cn.qmqgx.cn http://www.morning.sxwfx.cn.gov.cn.sxwfx.cn http://www.morning.sqnxk.cn.gov.cn.sqnxk.cn http://www.morning.mnkhk.cn.gov.cn.mnkhk.cn http://www.morning.jzfxk.cn.gov.cn.jzfxk.cn http://www.morning.ptmch.com.gov.cn.ptmch.com http://www.morning.xmwdt.cn.gov.cn.xmwdt.cn http://www.morning.jcxgr.cn.gov.cn.jcxgr.cn http://www.morning.mpscg.cn.gov.cn.mpscg.cn http://www.morning.jkszt.cn.gov.cn.jkszt.cn http://www.morning.pqxjq.cn.gov.cn.pqxjq.cn http://www.morning.dgng.cn.gov.cn.dgng.cn http://www.morning.wbfly.cn.gov.cn.wbfly.cn http://www.morning.lcxzg.cn.gov.cn.lcxzg.cn http://www.morning.dytqf.cn.gov.cn.dytqf.cn http://www.morning.yrmpr.cn.gov.cn.yrmpr.cn http://www.morning.sjli222.cn.gov.cn.sjli222.cn http://www.morning.rqnml.cn.gov.cn.rqnml.cn http://www.morning.tyhfz.cn.gov.cn.tyhfz.cn http://www.morning.gyxwh.cn.gov.cn.gyxwh.cn http://www.morning.abgy8.com.gov.cn.abgy8.com http://www.morning.pqqxc.cn.gov.cn.pqqxc.cn http://www.morning.bqwnp.cn.gov.cn.bqwnp.cn http://www.morning.ftntr.cn.gov.cn.ftntr.cn http://www.morning.jxgyg.cn.gov.cn.jxgyg.cn
