6758-Project / hockey

0 stars 0 forks source link

Generate Figures #41

Closed JakeColor closed 2 years ago

JakeColor commented 2 years ago

This utility generates the model quality charts described in M2 directions for an input list of predictions.

Helpful so we can mix & match models in the charts without repeatedly retraining them.

See the script docstring for usage.

TimkLee commented 2 years ago

What about moving both experiment_prediction_filenames = { "xgb_distance_and_angle_only": "./models/predictions/xgboost_distance_angle_only.csv" }

and

title = "Visual Summary - XGBoost Models" image_dir = "./figures/advanced_models/"

out and pass them in as a parameter generate_adv_model_figures(experiment_prediction_filenames, title = "Default", image_dir = "./"): ? This would make the code more reusable right?