Skip to content

Commit ba4f86d

Browse files
fabclmntjfsantos-dsarunnthevapalanmglcampos
authored
fix: Timegan loss training (#100)
Co-authored-by: Francisco Santos <36741643+jfsantos-ds@users.noreply.github.com> Co-authored-by: Arunn Thevapalan <arunn.work@gmail.com> Co-authored-by: Miguel Campos <mgl.mcampos@gmail.com>
1 parent 7d1bd13 commit ba4f86d

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
@@ -34,6 +34,9 @@ def make_net(model, n_layers, hidden_units, output_units, net_type='GRU'):
3434

3535

3636
class TimeGAN(BaseModel):
37+
38+
__MODEL__='TimeGAN'
39+
3740
def __init__(self, model_parameters, hidden_dim, seq_len, n_seq, gamma):
3841
self.seq_len=seq_len
3942
self.n_seq=n_seq

0 commit comments

Comments
 (0)