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
+ 5
5
Compare changes
  • Side-by-side
  • Inline
__author__ = "Lukas Leufen"
__date__ = '2021-02-'
from functools import reduce
from functools import reduce, partial
from mlair.model_modules import AbstractModelClass
from mlair.helpers import select_from_dict
@@ -62,8 +62,8 @@ class FCN(AbstractModelClass):
on the window_lead_time parameter.
"""
_activation = {"relu": keras.layers.ReLU(), "tanh": keras.layers.Activation("tanh"),
"sigmoid": keras.layers.Activation("sigmoid")}
_activation = {"relu": keras.layers.ReLU, "tanh": partial(keras.layers.Activation, "tanh"),
"sigmoid": partial(keras.layers.Activation, "sigmoid")}
_optimizer = {"adam": keras.optimizers.adam, "sgd": keras.optimizers.SGD}
_requirements = ["lr", "beta_1", "beta_2", "epsilon", "decay", "amsgrad", "momentum", "nesterov"]
@@ -125,9 +125,9 @@ class FCN(AbstractModelClass):
n_layer, n_hidden = self.layer_configuration
for layer in range(n_layer):
x_in = keras.layers.Dense(n_hidden)(x_in)
x_in = self.activation(x_in)
x_in = self.activation()(x_in)
x_in = keras.layers.Dense(self._output_shape)(x_in)
out = self.activation(x_in)
out = self.activation()(x_in)
self.model = keras.Model(inputs=x_input, outputs=[out])
def set_compile_options(self):
Loading