Fix: refactor getattr(keras.optimizers, optimizer) to `keras.optimi…

…zers.get(optimizer)` in autoencoders
equinor · Jun 24, 2024 · fe3530b · fe3530b
1 parent b64dc08
commit fe3530b
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/gordo/machine/model/factories/feedforward_autoencoder.py b/gordo/machine/model/factories/feedforward_autoencoder.py
@@ -88,7 +88,7 @@ class (e.x. Adam(lr=0.01,beta_1=0.9, beta_2=0.999)).  If no arguments are
 
     # Instantiate optimizer with kwargs
     if isinstance(optimizer, str):
-        Optim = getattr(keras.optimizers, optimizer)
+        Optim = keras.optimizers.get(optimizer)
         optimizer = Optim(**optimizer_kwargs)
 
     # Final output layer

diff --git a/gordo/machine/model/factories/lstm_autoencoder.py b/gordo/machine/model/factories/lstm_autoencoder.py
@@ -90,7 +90,7 @@ class (e.x. Adam(lr=0.01,beta_1=0.9, beta_2=0.999)).  If no arguments are
 
     # output layer
     if isinstance(optimizer, str):
-        Optim = getattr(keras.optimizers, optimizer)
+        Optim = keras.optimizers.get(optimizer)
         optimizer = Optim(**optimizer_kwargs)
 
     model.add(Dense(units=n_features_out, activation=out_func))