add models folder and create it if it doesnt exist

tryolabs · Jul 5, 2024 · 83e31ab · 83e31ab
1 parent 7715061
commit 83e31ab
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/challenge/model.py b/challenge/model.py
@@ -281,6 +281,8 @@ def fit(self, features: pd.DataFrame, target: pd.DataFrame) -> None:
         logging.info(f"Confusion Matrix:\n{conf_matrix}")
         logging.info(f"Classification Report:\n{class_report}")
 
+        # Ensure the directory exists
+        self.complete_model_path.parent.mkdir(parents=True, exist_ok=True)
         # Save the model
         model.save_model(self.complete_model_path)
         self._model = model

diff --git a/models/.gitkeep b/models/.gitkeep