Skip to content

Add verbosity TVAE (progress bar + save the loss values) #300

@npatki

Description

@npatki

Problem Description

Similar to #298, we should also add a verbosity option for TVAE that shows the user the fitting progress per epoch. After fitting, it should be possible to see the loss values.

Expected behavior

  1. Add a verbose parameter to TVAE (just like CTGAN). By default it can be False
  2. When set to True, increment a progress bar per (epoch x data iteration). Also show/refresh the loss value, similar to Add progress bar for CTGAN fitting (+ save the loss values) #298 (except there is only 1 loss value)
  3. Always save the loss values that correspond to the last fitting iteration. The columns can now be Epoch, Batch and Loss.
>>> model = TVAE(verbose=True, epochs=100)
>>> model.fit()
Fitting TVAE ...
Loss 0.024 |████████████████████████████         | 750/1000 [00:33<00:10, 229.00it/s]

# after modeling
>>> model.loss_values
   Epoch     Batch    Loss 
0      1     1        1.7863
1      1     2        1.5484
2      1     3        1.3633

Metadata

Metadata

Assignees

Labels

Type

No type

Projects

No projects

Milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions