fix: replace GumbelSoftmax by Softmax (#225)

ydataai · Jan 19, 2023 · 3d58b58 · 3d58b58
1 parent e75721f
commit 3d58b58
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/src/ydata_synthetic/synthesizers/regular/cramergan/model.py b/src/ydata_synthetic/synthesizers/regular/cramergan/model.py
@@ -187,9 +187,7 @@ def build_model(self, input_shape, dim, data_dim, activation_info: Optional[Name
         x = Dense(dim, activation='relu')(input_)
         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)
+        x = Dense(data_dim, activation='softmax')(x)
         return Model(inputs=input_, outputs=x)
 
 class Critic(tf.keras.Model):