ydataai / ydata-synthetic

Synthetic data generators for tabular and time-series data
https://docs.synthetic.ydata.ai
MIT License
1.42k stars 236 forks source link

Model training debugging capabilities #330

Open akashshah59 opened 7 months ago

akashshah59 commented 7 months ago

I was wondering if we could enable Model debugging and logging capabilites.

I've just been wanting to plot the loss curve for Generator and Discriminator, While I admire the design of how ydata has been built, it would be nice to have some tensor-board like plots epoch curves automatically flushed out, I've also been thinking of building visualizations do see how distribution shifts occur for Generator over time. I perused through base.py however, all I can see is training_params, which does have a log_frequency , but maybe we can add a /logging_dir as well.

I'd be willing to work on this, let me know what you guys think?