Model Training
==============

ModelTrainer
------------

.. automodule:: deepcausalmmm.core.trainer
   :members:
   :undoc-members:
   :show-inheritance:

Training Functions
------------------

.. automodule:: deepcausalmmm.core.train_model
   :members:
   :undoc-members:
   :show-inheritance: