如何为我经过训练的机器学习模型提供输入以预测输出?

时间:2021-03-08 06:19:15

标签: python python-3.x machine-learning data-science xgboost

我训练了一个机器学习模型来使用过去 5 年的数据预测番茄的价格。现在我想提供一些信息作为输入以获得预测输出,但我不知道该怎么做。这些是我用来训练模型区域名称、市场名称、品种、等级、Modal_Price_(Rs./Quintal)_log、年、月和日的特征。

#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))

均方根误差:0.285714

1 个答案:

答案 0 :(得分:-1)

如果你已经阅读了一个 csv 文件,你需要确保同名文件存在于同一个目录中。如果您正在使用 Pandas 创建数据框。

import pandas as pd
df = pd.read_csv('filename.csv')

如果您想提供原始输入,请参阅官方 docs