API Reference
=============

Models
------

Base Model
~~~~~~~~~~

.. automodule:: src.models.base_model
   :members:
   :undoc-members:
   :show-inheritance:

Poll Average Model
~~~~~~~~~~~~~~~~~~

.. automodule:: src.models.poll_average
   :members:
   :undoc-members:
   :show-inheritance:

Kalman Diffusion Model
~~~~~~~~~~~~~~~~~~~~~~~

.. automodule:: src.models.kalman_diffusion
   :members:
   :undoc-members:
   :show-inheritance:

Improved Kalman Model
~~~~~~~~~~~~~~~~~~~~~~

.. automodule:: src.models.improved_kalman
   :members:
   :undoc-members:
   :show-inheritance:

Hierarchical Bayes Model
~~~~~~~~~~~~~~~~~~~~~~~~~

.. automodule:: src.models.hierarchical_bayes
   :members:
   :undoc-members:
   :show-inheritance:

Utilities
---------

Data Utilities
~~~~~~~~~~~~~~

.. automodule:: src.utils.data_utils
   :members:
   :undoc-members:
   :show-inheritance:

Scripts
-------

Run All Models
~~~~~~~~~~~~~~

.. automodule:: src.scripts.run_all_models
   :members:
   :undoc-members:

Compare Models
~~~~~~~~~~~~~~

.. automodule:: src.scripts.compare_models
   :members:
   :undoc-members:

Generate Plots
~~~~~~~~~~~~~~

.. automodule:: src.scripts.generate_plots
   :members:
   :undoc-members:

Run All Pipeline
~~~~~~~~~~~~~~~~

.. automodule:: src.scripts.run_all
   :members:
   :undoc-members: