Parcourir la source

Fixed results writing

Nicholas Schense il y a 6 jours
Parent
commit
38117a2590
1 fichiers modifiés avec 1 ajouts et 1 suppressions
  1. 1 1
      train_model.py

+ 1 - 1
train_model.py

@@ -130,7 +130,7 @@ for run_num in range(config["training"]["ensemble_size"]):
 
     # Save test results by appending to the results file
     test_results_save_path = pl.Path(config["output"]["path"]) / f"results.json"
-    with open(test_results_save_path, "wr+") as f:
+    with open(test_results_save_path, "r+") as f:
         try:
             results = json.load(f)
         except json.JSONDecodeError: