|
|
@@ -28,5 +28,11 @@ object Main extends App {
|
|
|
println(stats.size)
|
|
|
|
|
|
val analysis = GamesAnalysis(teams_output, games_output, stats_output)
|
|
|
- GamesAnalysis.writeToCsv(analysis.merged, "output/merged_df")
|
|
|
+
|
|
|
+ Map(
|
|
|
+ "output/teams_games_df" -> analysis.teams_games.sort("game_id", "team_id"),
|
|
|
+ "output/stats_games_pts_df" -> analysis.stats_games_pts
|
|
|
+ .sort("game_id", "team_id"),
|
|
|
+ "output/final_df" -> analysis.merged.sort("game_id", "team_id")
|
|
|
+ ).foreachEntry((path, df) => GamesAnalysis.writeToCsv(df, path))
|
|
|
}
|