Skip to content

Commit a2c9a29

Browse files
authored
fix(TimeGAN): Restore define_gan call
1 parent ffec6b0 commit a2c9a29

1 file changed

Lines changed: 3 additions & 0 deletions

File tree

  • src/ydata_synthetic/synthesizers/timeseries/timegan

src/ydata_synthetic/synthesizers/timeseries/timegan/model.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -226,6 +226,9 @@ def get_batch_noise(self):
226226
.repeat())
227227

228228
def train(self, data, train_steps):
229+
# Assemble the model
230+
self.define_gan()
231+
229232
## Embedding network training
230233
autoencoder_opt = Adam(learning_rate=self.g_lr)
231234
for _ in tqdm(range(train_steps), desc='Emddeding network training'):

0 commit comments

Comments
 (0)