def train_model(team_stats, result_data, test_data):
# ??????
X, y = build_dataSet(team_stats, result_data)
# ??????
print("Fitting on %d game samples.." % len(X))
model = LogisticRegression()
model.fit(X, y)
#??10????????????
print("Doing cross-validation..")
print(cross_val_score(model, X, y, cv = 10, scoring='accuracy', n_jobs=-1).mean())
#??????model?????????
print('Predicting on test data..')
result = []
for index, row in test_data.iterrows():
team1 = row['Vteam']
team2 = row['Hteam']
pred = predict_winner(team1, team2, model, team_stats)
result.append(pred[0][0])
return result
评论列表
文章目录