Skip to content
Snippets Groups Projects

Resolve "release v1.4.0"

Merged Ghost User requested to merge release_v1.4.0 into master
1 file
+ 1
1
Compare changes
  • Side-by-side
  • Inline
@@ -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):
Loading