Models
======

BGM Family
----------

.. currentmodule:: bayesgm.models.bgm

BGM
^^^

.. autoclass:: BGM
   :no-undoc-members:

   .. rubric:: Training

   .. automethod:: fit

   .. rubric:: Inference

   .. automethod:: predict
   .. automethod:: generate
   .. automethod:: evaluate

   .. rubric:: Configuration

   .. automethod:: get_config

MNISTBGM
^^^^^^^^

.. autoclass:: MNISTBGM
   :no-undoc-members:

   .. rubric:: Training

   .. automethod:: fit

   .. rubric:: Inference

   .. automethod:: predict
   .. automethod:: generate
   .. automethod:: evaluate

   .. rubric:: Configuration

   .. automethod:: get_config


CausalBGM Family
----------------

.. currentmodule:: bayesgm.models.causalbgm

CausalBGM
^^^^^^^^^

.. autoclass:: CausalBGM
   :no-undoc-members:

   .. rubric:: Training

   .. automethod:: fit

   .. rubric:: Inference

   .. automethod:: predict
   .. automethod:: evaluate

   .. rubric:: Configuration

   .. automethod:: get_config

IdentifiableCausalBGM
^^^^^^^^^^^^^^^^^^^^^

.. autoclass:: IdentifiableCausalBGM
   :no-undoc-members:

   .. rubric:: Training

   .. automethod:: fit

   .. rubric:: Inference

   .. automethod:: predict
   .. automethod:: evaluate

   .. rubric:: Configuration

   .. automethod:: get_config

FullMCMCCausalBGM
^^^^^^^^^^^^^^^^^

.. autoclass:: FullMCMCCausalBGM
   :no-undoc-members:

   .. rubric:: Training

   .. automethod:: fit
   .. automethod:: run_mcmc_training

   .. rubric:: Inference

   .. automethod:: predict
   .. automethod:: evaluate

   .. rubric:: Configuration

   .. automethod:: get_config