Head Stabilization
==================

Model
-----
.. autoclass:: flygym.examples.head_stabilization.JointAngleScaler
    :members:
    :undoc-members:
    :show-inheritance:
    :inherited-members:

.. autoclass:: flygym.examples.head_stabilization.WalkingDataset
    :members:
    :undoc-members:
    :show-inheritance:

.. autoclass:: flygym.examples.head_stabilization.ThreeLayerMLP
    :members:
    :show-inheritance:

.. autoclass:: flygym.examples.head_stabilization.HeadStabilizationInferenceWrapper
    :members:
    :undoc-members:
    :show-inheritance:
    :inherited-members:


Utilities
---------
.. autofunction:: flygym.examples.head_stabilization.util.get_head_stabilization_model_paths