小编Gra*_*ton的帖子

如何将多维数据传入xgboost模型

本质上这就是我对 xgboost 的看法

model = XGBClassifier()

model.fit(X_train, y_train)
Run Code Online (Sandbox Code Playgroud)

其中X_trainy_train是 numpy 数组。我的问题是,X_train似乎必须采用数字矩阵的格式,其中每行都是一组数字,例如:

[1, 5, 3, 6]
Run Code Online (Sandbox Code Playgroud)

然而,我拥有的数据是一组向量的格式。每个向量由 1、3 之间的数字和 0、1 之间的置信区间组成。因此 my 的一行X_train看起来像:

[[1, .84], [2, .5], [3, .44], [2, .76]]
Run Code Online (Sandbox Code Playgroud)

但是我不知道如何将这种格式的数据传递给 xgboost。我对 xgboost 相当陌生,所以我一直在阅读文档,但我似乎找不到我要找的东西。谢谢你的帮助。

python machine-learning xgboost

2
推荐指数
1
解决办法
5816
查看次数

标签 统计

machine-learning ×1

python ×1

xgboost ×1