Skip to content

Plots training losses and metrics in Jupyter while training your `keras` model.

License

Notifications You must be signed in to change notification settings

ig248/kerashistoryplot

Repository files navigation

PyPI version Build Status Coverage Status

History plot for Keras model training

history plot with batches

Plots training losses and metrics in Jupyter while training your keras model.

This project started as an attempt to extend livelossplot, but I realized that a rewrite was easier. Key improvements are:

  • keeps full history, including losses for each batch in each epoch
  • plots scatter of batch losses
  • plots arbitrary log entries, such a lr (learning rate)
  • simplifies codebase extendibility

Basic Usage

Install package from source:

pip install git+https://github.com/ig248/kerashistoryplot

Or from PyPi:

pip install kerashistoryplot

Import function:

from livehistoryplot.callbacks import PlotHistory

Use in callbacks:

model.fit(
    X, y,
    epochs=20,
    callbacks=[
        ReduceLROnPlateau(),
        PlotHistory(batches=True, n_cols=3, figsize=(15, 7))
    ]
)

More Examples

See notebooks/

Tests

Run

make dev-install
make test

References

Original livelossplot: https://github.com/stared/livelossplot

About

Plots training losses and metrics in Jupyter while training your `keras` model.

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published