Fix data paths

2024-03-07 15:25:44 +02:00 · 2024-03-07 15:25:44 +02:00 · 5f1ff8a93d
commit 5f1ff8a93d
parent 33f03671b6
1 changed files with 3 additions and 0 deletions
--- a/model.py
+++ b/model.py
@ -94,6 +94,9 @@ def train_model():
    model = LinearRegression()
    model.fit(x_train, y_train)

+    # create the model's parent directory if it doesn't exist
+    os.makedirs(os.path.dirname(model_file_path), exist_ok=True)
+
    # Save the trained model to a file
    with open(model_file_path, "wb") as f:
        pickle.dump(model, f)