Update src/train_model.py

src/train_model.py

@@ -21,7 +21,7 @@ Synthetic Data Strategy:
 """
 import os
 
-BASE_DIR = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
+BASE_DIR = os.path.dirname(os.path.abspath(__file__))
 DATA_DIR = os.path.join(BASE_DIR, "data")
 
 import torch
@@ -39,7 +39,7 @@ from src.disaster_predictors import (
     FLOOD_FEATURES, CYCLONE_FEATURES, LANDSLIDE_FEATURES, EARTHQUAKE_FEATURES
 )
 
-MODEL_DIR = os.path.join(BASE_DIR, "models")
+MODEL_DIR = "models"
 SEED = 42
 np.random.seed(SEED)
 torch.manual_seed(SEED)