【发布时间】:2021-06-06 01:23:37
【问题描述】:
我已经训练了一个机器学习模型来使用过去 5 年的数据来预测番茄的价格。现在我想提供一些信息作为输入来获得预测的输出,但我不知道该怎么做。这些是我用来训练模型 District Name、Market Name、Variety、Grade、Modal_Price_(Rs./Quintal)_log、year、month 和 day 的特征。
#XGBoost
xgbm = XGBRegressor()
xgbm.fit(x_train,y_train)
preds = xgbm.predict(x_test)
rmse = np.sqrt(mean_squared_error(y_test, preds))
print("RMSE: %f" % (rmse))
RMSE:0.285714
【问题讨论】:
标签: python python-3.x machine-learning data-science xgboost