Skip to content
Snippets Groups Projects
Commit 24808a64 authored by leufen1's avatar leufen1
Browse files

use mse loss for rnn

parent f384b448
No related branches found
No related tags found
5 merge requests!319add all changes of dev into release v1.4.0 branch,!318Resolve "release v1.4.0",!317enabled window_lead_time=1,!295Resolve "data handler FIR filter",!259Draft: Resolve "WRF-Datahandler should inherit from SingleStationDatahandler"
Pipeline #68570 passed
Stage: test
Stage: docs
Stage: pages
Stage: deploy
Show whitespace changes
Inline Side-by-side
mlair/model_modules/recurrent_networks.py
+ 1
− 1
View file @ 24808a64
...@@ -136,7 +136,7 @@ class RNN(AbstractModelClass): ...@@ -136,7 +136,7 @@ class RNN(AbstractModelClass):
raise AttributeError(f"Given activation {activation} is not supported in this model class.") raise AttributeError(f"Given activation {activation} is not supported in this model class.")
def set_compile_options(self): 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]} "metrics": ["mse", "mae", var_loss]}
def _set_optimizer(self, optimizer, **kwargs): def _set_optimizer(self, optimizer, **kwargs):
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment