Update BaseModel.py

BaseModel.py

@@ -47,12 +47,12 @@ class GeoMagModel:
 
         self.model.fit(df)
 
-    def forecast(self, periods=6, freq='h'):
+    def forecast(self, periods=1, freq='h'):
         """
         Make a future forecast with the trained model.
         
         Args:
-            periods (int): The number of future periods to forecast. Defaults to 6.
+            periods (int): The number of future periods to forecast. Defaults to 1.
             freq (str): The frequency of the forecast ('h' for hours). Defaults to 'h'.
         
         Returns:
@@ -65,7 +65,7 @@ class GeoMagModel:
         forecast = self.model.predict(future_dates)
         return forecast[['ds', 'yhat', 'yhat_lower', 'yhat_upper']]
 
-    def cross_validate(self, initial='384 hours', period='12 hours', horizon='6 hours'):
+    def cross_validate(self, initial='36 hours', period='12 hours', horizon='1 hours'):
         """
         Perform cross-validation on the trained Prophet model.