added pytorch forecasting wrapper to make it easier

rvandewater · Oct 31, 2023 · d03deed · d03deed
1 parent 119dbcf
commit d03deed
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/configs/tasks/BinaryClassification.gin b/configs/tasks/BinaryClassification.gin
@@ -16,7 +16,7 @@ train_common.ram_cache = True
 
 # DEEP LEARNING
 DLPredictionWrapper.loss = @cross_entropy
-
+DLPredictionPytorchForecastingWrapper.loss= @cross_entropy
 # SELECTING PREPROCESSOR
 preprocess.preprocessor = @base_classification_preprocessor
 preprocess.vars = %vars

diff --git a/configs/tasks/Regression.gin b/configs/tasks/Regression.gin
@@ -16,6 +16,7 @@ train_common.ram_cache = True
 
 # LOSS FUNCTION
 DLPredictionWrapper.loss = @mse_loss
+DLPredictionPytorchForecastingWrapper.loss = @mse_loss
 MLWrapper.loss = @mean_squared_error
 
 # SELECTING PREPROCESSOR