xgboost进行多输出回归预测

男娘i 2023-02-24 07:57 15阅读 0赞

如果我们需要用到多个维度的预测，也就是需要预测 很多个变量的时候

又不想训练很多次xg（假设有一个待预测标签就训练一次），因为 这样代码看起来也很繁琐

我们可以使用sklearn来 完成。

sklearn提供了一个包装类MultiOutputRegressor，这里我使用的xgboost进行预测。

思路很简单，其实就是根据标签依次进行预测。

代码如下：

from sklearn.multioutput import MultiOutputRegressor
    import xgboost  as xgb
    import  pandas  as  pd
    from sklearn.metrics import r2_score
    from sklearn.model_selection import train_test_split
    
    #获取数据
    df=pd.read_csv('XXX.txt',sep='\t')
    X=df.drop(['user_id','y1','y2'],axis=1)
    y=df[['y1','y2']]
    train_X,test_X,train_y,test_y = train_test_split(X,y,test_size=0.3, random_state=4)
    #准备参数
    other_params = {'learning_rate': 0.1, 'n_estimators': 300, 'max_depth': 5, 'min_child_weight': 1, 'seed': 0, 'subsample': 0.8, 'colsample_bytree': 0.8, 'gamma': 0, 'reg_alpha': 0, 'reg_lambda': 1}
    multioutputregressor = MultiOutputRegressor(xgb.XGBRegressor(objective='reg:squarederror',**other_params)).fit(train_X, train_y)
    check=multioutputregressor.predict(test_X)
    print(check)

可以看到，这里的代码仅仅使用两个y（y1,y2），要是根据我们不同的需求，我们可以使用很多个y就可以，比如（y1,y2,y3,...,yn）

发表评论取消回复

表情：

评论列表（有 0 条评论，15人围观）

还没有评论，来说两句吧...

相关阅读

相关使用布谷鸟算法优化的LSTM进行数据回归预测

使用布谷鸟算法优化的LSTM进行数据回归预测在时间序列数据分析方面，长短时记忆网络（LSTM）已经成为一种流行的方法。然而，LSTM模型中的大量参数使得该模型的训练变得困难

水深无声/ 2024年03月17日 16:46/ 0 赞/ 135 阅读

相关使用Python进行多目标回归

使用Python进行多目标回归多目标回归是指模型需要同时预测多个输出变量的情况，也被称为多输出回归。在这种情况下，我们的数据集包含多个目标变量，而不是单个目标变量。本文将介

àì夳堔傛蜴生んèń/ 2024年03月17日 13:19/ 0 赞/ 25 阅读

相关使用MATLAB进行多元非线性回归拟合预测

使用MATLAB进行多元非线性回归拟合预测在数据分析和预测建模中，回归分析是一种常用的统计方法。当我们需要预测一个或多个响应变量与多个预测变量之间的关系时，多元非线性回归是

「爱情、让人受尽委屈。」/ 2024年03月04日 02:15/ 0 赞/ 138 阅读

相关线性回归预测房价

> 给定数据集dataSet，每一行代表一组数据记录,每组数据记录中，第一个值为房屋面积（单位：平方英尺），第二个值为房屋中的房间数，第三个值为房价（单位：千美元），试用梯度下

骑猪看日落/ 2023年06月11日 11:23/ 0 赞/ 68 阅读

相关 xgboost进行多输出回归预测

如果我们需要用到多个维度的预测，也就是需要预测很多个变量的时候又不想训练很多次xg（假设有一个待预测标签就训练一次），因为这样代码看起来也很繁琐我们可以使用skle

男娘i/ 2023年02月24日 07:57/ 0 赞/ 16 阅读

相关多元logit回归参数估计（多分类logit回归预测）

目录多元离散选择模型简介样例程序样例程序的问题问题解决多元离散选择模型简介常用的离散选择模型有logit模型及probit模型，其

桃扇骨/ 2022年09月25日 11:20/ 0 赞/ 625 阅读

相关使用 XGBoost 进行时间序列预测，人口出生预测迷你教程

文章目录内容介绍 XGBoost 集成时间序列数据准备用于时间序列预测的 XGBoost 内容介绍 XGBoost 是针对分类和回归

素颜马尾好姑娘i/ 2022年09月13日 14:23/ 0 赞/ 1099 阅读

相关 Scikit中的特征选择，XGboost进行回归预测，模型优化的实战

前天偶然在一个网站上看到一个数据分析的比赛（[sofasofa][]），自己虽然学习一些关于机器学习的内容，但是并没有在比赛中实践过，于是我带着一种好奇心参加了这次比赛。 >

小咪咪/ 2022年05月29日 12:13/ 0 赞/ 211 阅读

相关回归预测评估指标

标注说明 ff表示预测值，yy表示实际值评价指标 MAE(Mean Absolute Error) 平均绝对误差 MAE

青旅半醒/ 2022年05月27日 09:16/ 0 赞/ 296 阅读

相关天池优惠卷预测--XGBoost训练模型

训练集经过上一个步骤之后，我们可以得到训练的特征数据集，这里一共有两个数据集，按日期分的，接下来利用训练集里得到模型。 XGBoost模型这里讲解利用XGBo

r囧r小猫/ 2022年05月15日 05:44/ 0 赞/ 296 阅读