Visualization Documentation
====================================

Dataset exploration
------------------------------

.. currentmodule:: supernnova.visualization.visualize.py

.. automodule:: supernnova.visualization.visualize
    :members:
    :show-inheritance:

Plotting predictions
------------------------------

.. currentmodule:: supernnova.visualization.early_prediction.py

.. automodule:: supernnova.visualization.early_prediction
    :members:
    :show-inheritance:

.. currentmodule:: supernnova.visualization.prediction_distribution.py

.. automodule:: supernnova.visualization.prediction_distribution
    :members:
    :show-inheritance: