Add interactive training loss visualization#6
Open
Vyomkesh13 wants to merge 9 commits into
Open
Conversation
…d autoencoder contributions
This feature introduces interactive HTML plots for training and validation loss, replacing static PDF diagnostics. Scientists can now zoom into specific epoch ranges, hover for exact loss values, and toggle traces on/off. Key improvements: - Interactive controls (zoom, pan, hover) - Self-contained HTML output (shareable via email) - Uses existing loss_data.npy format (backward compatible) - Professional Plotly styling suitable for publications Implementation: - New module: baler/modules/interactive_loss_plot.py - Comprehensive tests: tests/test_interactive_loss_plot.py (15+ cases) - Usage examples: examples/example_loss_plots.py - Documentation: docs/INTERACTIVE_LOSS_PLOTS.md Dependencies: Adds plotly>=5.0.0 Testing: All tests pass, no regressions in existing test suite.
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
2 participants
Add this suggestion to a batch that can be applied as a single commit.This suggestion is invalid because no changes were made to the code.Suggestions cannot be applied while the pull request is closed.Suggestions cannot be applied while viewing a subset of changes.Only one suggestion per line can be applied in a batch.Add this suggestion to a batch that can be applied as a single commit.Applying suggestions on deleted lines is not supported.You must change the existing code in this line in order to create a valid suggestion.Outdated suggestions cannot be applied.This suggestion has been applied or marked resolved.Suggestions cannot be applied from pending reviews.Suggestions cannot be applied on multi-line comments.Suggestions cannot be applied while the pull request is queued to merge.Suggestion cannot be applied right now. Please check back later.
Summary
This PR adds interactive HTML visualization for BALER training and validation
loss, providing a modern alternative to static PDF diagnostics.
The new plots allow users to zoom, pan, hover for exact values, and toggle
training/validation traces, making it easier to analyze training convergence.
Key Features
loss_data.npyformatImplementation
baler/modules/interactive_loss_plot.pytests/test_interactive_loss_plot.pyexamples/example_loss_plots.pydocs/INTERACTIVE_LOSS_PLOTS.mdDependencies
plotly>=5.0.0Testing
torchdependency;full test suite is expected to pass in CI environment