索引器错误:元组索引超出范围(XGBoost)

2024-09-28 23:16:59 发布

您现在位置:Python中文网/ 问答频道 /正文

我试图预测一些商店的每周销售额。我想使用XGBoost,但我一直遇到以下错误:

IndexError: tuple index out of range

我的代码:

from sklearn.model_selection import train_test_split

features = ["Store",'Dept','Temperature','Fuel_Price','CPI','Unemployment']

y = allmergecopy.Weekly_Sales
X = allmergecopy[features]
X_train, y_train, X_valid, y_valid = train_test_split(y,X,test_size=.2, train_size=.8)

from xgboost import XGBRegressor

my_model = XGBRegressor(n_estimators=100)
my_model.fit(X_train, y_train)

供参考:

y_train.shape = (84314,), X_train.shape = (337256,)


Tags: fromtestimportsizemodelmytrain商店
2条回答

而不是:

X_train, y_train, X_valid, y_valid = train_test_split(y,X,test_size=.2, train_size=.8)

写:

x_train,x_valid,y_train,y_valid= train_test_split(x,y,test_size=0.2)

然后使用:

from xgboost import XGBRegressor
my_model = XGBRegressor(n_estimators=100)
my_model.fit(x_train, y_train)

使用X作为:

X=allmergecopy.drop("Weekly_Sales",axis=1)

相关问题 更多 >