search

stared / livelossplot

Live training loss plot in Jupyter Notebook for Keras, PyTorch and others
https://p.migdal.pl/livelossplot
MIT License
1.29k stars 143 forks source link

Feature request: smoothing line to show trend #65

Closed chithangduong closed 5 years ago

chithangduong commented 5 years ago

It would be great if there is a trend line by smoothing the data points. This is especially useful for training GAN as the loss is usually noisy.

stared commented 5 years ago

@chithangduong There is a functionality to write custom subplots: https://github.com/stared/livelossplot/blob/master/livelossplot/matplotlib_subplots.py, vide https://github.com/stared/livelossplot/blob/master/examples/2d_prediction_maps.ipynb. Please create one with smooth, and show a working example (as a Pull Request).

It if works well, I will merge that (after a code review).