fix: Deactivate GumbelSoftmax. (#184)

Co-authored-by: fabiana <[email protected]>

Author: fabclmnt

src/ydata_synthetic/synthesizers/regular/dragan/model.py

@@ -185,6 +185,6 @@ def build_model(self, input_shape, dim, data_dim, activation_info: NamedTuple =
         x = Dense(dim * 2, activation='relu')(x)
         x = Dense(dim * 4, activation='relu')(x)
         x = Dense(data_dim)(x)
-        if activation_info:
-            x = GumbelSoftmaxActivation(activation_info, tau=tau)(x)
+        #if activation_info:
+        #    x = GumbelSoftmaxActivation(activation_info, tau=tau)(x)
         return Model(inputs=input, outputs=x)

src/ydata_synthetic/synthesizers/regular/wgan/model.py

@@ -167,8 +167,8 @@ def build_model(self, input_shape, dim, data_dim, activation_info: Optional[Name
         x = Dense(dim * 2, activation='relu')(x)
         x = Dense(dim * 4, activation='relu')(x)
         x = Dense(data_dim)(x)
-        if activation_info:
-            x = GumbelSoftmaxActivation(activation_info, tau=tau)(x)
+        #if activation_info:
+        #    x = GumbelSoftmaxActivation(activation_info, tau=tau)(x)
         return Model(inputs=input, outputs=x)
 
 class Critic(tf.keras.Model):

src/ydata_synthetic/synthesizers/regular/wgangp/model.py

@@ -184,8 +184,8 @@ def build_model(self, input_shape, dim, data_dim, activation_info: Optional[Name
         x = Dense(dim * 2, activation='relu')(x)
         x = Dense(dim * 4, activation='relu')(x)
         x = Dense(data_dim)(x)
-        if activation_info:
-            x = GumbelSoftmaxActivation(activation_info, tau=tau)(x)
+        #if activation_info:
+        #    x = GumbelSoftmaxActivation(activation_info, tau=tau)(x)
         return Model(inputs=input, outputs=x)
 
 class Critic(tf.keras.Model):