astrodata.ml.models package
===========================

Submodules
----------

astrodata.ml.models.BaseMlModel module
--------------------------------------

.. automodule:: astrodata.ml.models.BaseMlModel
   :members:
   :undoc-members:
   :show-inheritance:

astrodata.ml.models.PytorchModel module
---------------------------------------

.. automodule:: astrodata.ml.models.PytorchModel
   :members:
   :undoc-members:
   :show-inheritance:

astrodata.ml.models.SklearnModel module
---------------------------------------

.. automodule:: astrodata.ml.models.SklearnModel
   :members:
   :undoc-members:
   :show-inheritance:

astrodata.ml.models.TensorflowModel module
------------------------------------------

.. automodule:: astrodata.ml.models.TensorflowModel
   :members:
   :undoc-members:
   :show-inheritance:

astrodata.ml.models.XGBoostModel module
---------------------------------------

.. automodule:: astrodata.ml.models.XGBoostModel
   :members:
   :undoc-members:
   :show-inheritance:

Module contents
---------------

.. automodule:: astrodata.ml.models
   :members:
   :undoc-members:
   :show-inheritance: