python - 如何使用预测模型 python 预测列中的特定行？

Question

我有以下有效的代码:

for model in models:
    model.fit(trainset)
    predictions = model.test(testset)
    top_n = get_top_n(predictions, n=5)

    # Print the first one
    user = list(top_n.keys())[0]
    print(f'model: {model}, {user}: {top_n[user]}')
print('Top N computation successful!')

但挑战在于 - 我正在尝试使用该模型来预测足球比赛。感兴趣的两列是:home_game 和 away_game。

这些列中感兴趣的两个项目是:Chelsea 和 West Ham。

使用上述模型有条件地选择这两项的最佳方法是什么？

Answer 1

我假设您的数据框看起来像这样:

df = pd.DataFrame({'home_game':['Man Utd', 'Chelsea', 'Arsenal'], 'away_game':['Liverpool','West Ham','Spurs']})

    away_game   home_game
0   Liverpool   Man Utd
1   West Ham    Chelsea
2   Spurs       Arsenal

您可以这样选择切尔西和西汉姆的行:

test = df[(df['home_game'] == 'Chelsea') & (df['away_game'] == 'West Ham')]

并通过传递结果数据帧来预测这一点(取决于您的模型预测函数):

model.predict(test)