From 24808a64f4ae1d1cadb5237b2a19ead86233a445 Mon Sep 17 00:00:00 2001
From: leufen1 <l.leufen@fz-juelich.de>
Date: Wed, 26 May 2021 14:23:13 +0200
Subject: [PATCH] use mse loss for rnn
---
mlair/model_modules/recurrent_networks.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/mlair/model_modules/recurrent_networks.py b/mlair/model_modules/recurrent_networks.py
index 55a3d585..7adc9111 100644
--- a/mlair/model_modules/recurrent_networks.py
+++ b/mlair/model_modules/recurrent_networks.py
@@ -136,7 +136,7 @@ class RNN(AbstractModelClass):
raise AttributeError(f"Given activation {activation} is not supported in this model class.")
def set_compile_options(self):
- self.compile_options = {"loss": [custom_loss([keras.losses.mean_squared_error, var_loss])],
+ self.compile_options = {"loss": [keras.losses.mean_squared_error],
"metrics": ["mse", "mae", var_loss]}
def _set_optimizer(self, optimizer, **kwargs):
--
GitLab