我试图预测一些商店的每周销售额。我想使用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,)
而不是:
写:
然后使用:
使用X作为:
X=allmergecopy.drop("Weekly_Sales",axis=1)
相关问题 更多 >
编程相关推荐