Skip to content

Commit 3d58b58

Browse files
authored
fix: replace GumbelSoftmax by Softmax (#225)
1 parent e75721f commit 3d58b58

1 file changed

Lines changed: 1 addition & 3 deletions

File tree

  • src/ydata_synthetic/synthesizers/regular/cramergan

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

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -187,9 +187,7 @@ def build_model(self, input_shape, dim, data_dim, activation_info: Optional[Name
187187
x = Dense(dim, activation='relu')(input_)
188188
x = Dense(dim * 2, activation='relu')(x)
189189
x = Dense(dim * 4, activation='relu')(x)
190-
x = Dense(data_dim)(x)
191-
if activation_info:
192-
x = GumbelSoftmaxActivation(activation_info, tau=tau)(x)
190+
x = Dense(data_dim, activation='softmax')(x)
193191
return Model(inputs=input_, outputs=x)
194192

195193
class Critic(tf.keras.Model):

0 commit comments

Comments
 (0)