temporal

aimclub · May 15, 2024 · 30bc2b3 · 30bc2b3
1 parent d9ac125
commit 30bc2b3
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/fedot_ind/core/repository/model_repository.py b/fedot_ind/core/repository/model_repository.py
@@ -177,7 +177,9 @@ class AtomizedModel(Enum):
         'stl_arima': STLForecastARIMAImplementation,
         'ets': ExpSmoothingImplementation,
         'cgru': CGRUImplementation,
-        'glm': GLMIndustrial
+        'glm': GLMIndustrial,
+        # variational
+        'deepar_model': DeepAR,
     }
 
     FORECASTING_PREPROC = {