Skip to content

C
covid_analysis
Commit d38664a2 authored by Joaquin Torres's avatar Joaquin Torres
Browse files
Options

Generated fitted models

parent 6b0cf754
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 1 deletion
explicability/fit_final_models.py
View file @ d38664a2
...@@ -144,4 +144,5 @@ if __name__ == "__main__": ...@@ -144,4 +144,5 @@ if __name__ == "__main__":
model_save_path = f"./output/fitted_models/{group}_{method_names[j]}_{model_choices[method_name]}.pkl" model_save_path = f"./output/fitted_models/{group}_{method_names[j]}_{model_choices[method_name]}.pkl"
# Save the model to disk # Save the model to disk
with open(model_save_path, 'wb') as f: with open(model_save_path, 'wb') as f:
pickle.dump(model, f) pickle.dump(fitted_model, f)
\ No newline at end of file print('Successful fitting')
\ No newline at end of file
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment