reduced estimators for RandomForest test case

unit8co · Aug 10, 2022 · 3c7b9a4 · 3c7b9a4
1 parent 3baf4b8
commit 3c7b9a4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/darts/tests/models/forecasting/test_local_forecasting_models.py b/darts/tests/models/forecasting/test_local_forecasting_models.py
@@ -52,7 +52,7 @@
     (NaiveSeasonal(), 32.4),
     (KalmanForecaster(dim_x=3), 17.0),
     (LinearRegressionModel(lags=12), 11.0),
-    (RandomForest(lags=12, n_estimators=200, max_depth=3), 15.5),
+    (RandomForest(lags=12, n_estimators=5, max_depth=3), 17.0),
 ]
 
 # forecasting models with exogenous variables support