Neural Network Module
======================

The neural network module provides the ``LightweightNN`` class for building and training neural networks.

.. automodule:: fivedreg.neural_network
   :members:
   :undoc-members:
   :show-inheritance:

LightweightNN Class
-------------------

.. autoclass:: fivedreg.neural_network.LightweightNN
   :members:
   :undoc-members:
   :show-inheritance:
   :special-members: __init__
   :no-index: