Created models.py for storing model classes that take parameter kwargs as input

source/hypermodels.py

@@ -439,7 +439,3 @@ class SEQ_2_SEQ_MMVAE(HyperModel):
 
     def build(self, hp):
         pass
-
-
-# TODO:
-#     1) Refactor VAE losses to be proper losses, using K.add_loss() instead of global variables!

source/models.py

+# @author lucasmiranda42
+
+from keras import backend as K
+from tensorflow.keras import Input, Model, Sequential
+from tensorflow.keras.constraints import Constraint, UnitNorm
+from tensorflow.keras.layers import Bidirectional, Dense, Dropout
+from tensorflow.keras.layers import Lambda, Layer, LSTM
+from tensorflow.keras.layers import RepeatVector, TimeDistributed
+from tensorflow.keras.losses import Huber
+from tensorflow.keras.optimizers import Adam
+import tensorflow as tf
+
+
+class SEQ_2_SEQ_AE:
+    pass
+
+
+class SEQ_2_SEQ_VAE:
+    pass
+
+
+class SEQ_2_SEQ_MVAE(HyperModel):
+    pass
+
+
+class SEQ_2_SEQ_MMVAE(HyperModel):
+    pass