| from pathlib import Path |
| import csv |
|
|
| from la_liga_score_predictor import LaLigaScorePredictor |
|
|
| predictor = LaLigaScorePredictor.from_defaults( |
| dataset_csv_path="sample_history.csv" |
| ) |
|
|
| input_path = Path("sample_fixtures.csv") |
| output_path = Path("predictions_output.csv") |
|
|
| with input_path.open(newline="", encoding="utf-8") as src, output_path.open("w", newline="", encoding="utf-8") as dst: |
| reader = csv.DictReader(src) |
| fieldnames = ["home_team", "away_team", "match_date", "predicted_score", "home_win", "draw", "away_win", "confidence_level"] |
| writer = csv.DictWriter(dst, fieldnames=fieldnames) |
| writer.writeheader() |
| for row in reader: |
| result = predictor.predict_match( |
| home_team=row["home_team"], |
| away_team=row["away_team"], |
| match_date=row["match_date"], |
| ) |
| writer.writerow( |
| { |
| "home_team": row["home_team"], |
| "away_team": row["away_team"], |
| "match_date": row["match_date"], |
| "predicted_score": result["predicted_score"], |
| "home_win": result["result_probabilities"]["home_win"], |
| "draw": result["result_probabilities"]["draw"], |
| "away_win": result["result_probabilities"]["away_win"], |
| "confidence_level": result["confidence_level"], |
| } |
| ) |
|
|
| print(f"Wrote {output_path}") |
|
|
