Как предсказать конкретную строку в столбце, используя модель прогнозирования 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)