如何仅在训练时使用所有特征,但在测试时仅使用2个特征?

我正在使用这个数据集构建一个预测英超联赛(足球)结果的机器学习模型,该数据集包含主队进球数、客队进球数、射正球门次数等特征。以下是我在加载数据集后的当前代码:

features = list(data.columns.values)X, y = data[features], data.FTR     #FTR代表全场比赛结果print(X.shape)  -> (4940, 20)X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2, random_state=18)nb = GaussianNB()nb.fit(X_train, y_train)y_nb = nb.predict(X_test)

这给出了非常高的准确率(72%),但这是因为当我要求模型预测结果时,我给它提供了我想预测的比赛的统计数据(包括进球数)。有没有办法“隐藏”除主队和客队之外的所有特征,并以这种方式预测结果?

我尝试过这样做:

 X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2, random_state=18) X_test = X_test.iloc[:, [0, 1]] #这只保留了主队名称和客队名称的列 nb = GaussianNB() nb.fit(X_train, y_train) y_nb = nb.predict(X_test)

然而,这会导致以下错误:

ValueError: operands could not be broadcast together with shapes (988,2) (20,) 

回答:

如果你想保留你的特征提供的所有信息,可以考虑使用平均值或某种比赛前历史统计数据的度量来训练你的模型。即,如果A队在训练比赛前三场比赛中分别进了2、3和1个球,而在训练的那场比赛中进了5个球,可以使用前三场比赛的平均值而不是实际的进球总数。你的训练误差可能会更高,但当你去预测一场新比赛时,你仍然可以使用尽可能多的数据。

正如@***所说,当你试图训练一个作为预测器的模型时,仅使用在你进行预测时可用的变量。

Related Posts

如何在仅有一个特征的情况下绘制SVM超平面

我有一个只包含一个特征的数据集,我使用scikit-l…

将在Python中训练的XGBoost模型移植到用C/C++编写的另一个系统

假设我已经在Python中成功训练了一个XGBoost…

代码中出现的问题 “无法将字符串转换为浮点数”

我正在通过一个GitHub链接学习线性回归 &#822…

Google Cloud Vision AI的替代方案

已关闭。 此问题正在寻求关于软件库、教程、工具、书籍或…

绘制XGBoost模型特征重要性的增益、覆盖率和权重

我有一个XGBoost模型 xgboost_model…

评估一个ANN线性回归模型

我用Keras创建了我的第一个ANN。这是一个具有5个…

发表回复

您的邮箱地址不会被公开。 必填项已用 * 标注