OmniSafe Model-based Model
==========================

.. currentmodule:: omnisafe.algorithms.model_based.base.ensemble

Standard Scaler
---------------

.. card::
    :class-header: sd-bg-success sd-text-white
    :class-card: sd-outline-success  sd-rounded-1

    Documentation
    ^^^

    .. autoclass:: StandardScaler
        :members:
        :private-members:

Initialize Weight
-----------------

.. card::
    :class-header: sd-bg-success sd-text-white
    :class-card: sd-outline-success  sd-rounded-1

    Documentation
    ^^^

    .. autofunction:: init_weights

Unbatched Forward
-----------------

.. card::
    :class-header: sd-bg-success sd-text-white
    :class-card: sd-outline-success  sd-rounded-1

    Documentation
    ^^^

    .. autofunction:: unbatched_forward

Ensemble Fully-Connected Layer
------------------------------

.. card::
    :class-header: sd-bg-success sd-text-white
    :class-card: sd-outline-success  sd-rounded-1

    Documentation
    ^^^

    .. autoclass:: EnsembleFC
        :members:
        :private-members:

Ensemble Model
--------------

.. card::
    :class-header: sd-bg-success sd-text-white
    :class-card: sd-outline-success  sd-rounded-1

    Documentation
    ^^^

    .. autoclass:: EnsembleModel
        :members:
        :private-members:


Ensemble Dynamics Model
-----------------------

.. card::
    :class-header: sd-bg-success sd-text-white
    :class-card: sd-outline-success  sd-rounded-1

    Documentation
    ^^^

    .. autoclass:: EnsembleDynamicsModel
        :members:
        :private-members:
