bayesian_models.rst 520 字节
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15
The ``models`` module
======================

This module contains the two Bayesian Models available in this library, namely
the bayesian version of the Wide and TabMlp models, referred as ``BayesianWide``
and ``BayesianTabMlp``


.. autoclass:: pytorch_widedeep.bayesian_models.tabular.bayesian_linear.bayesian_wide.BayesianWide
    :exclude-members: forward
    :members:

.. autoclass:: pytorch_widedeep.bayesian_models.tabular.bayesian_mlp.bayesian_tab_mlp.BayesianTabMlp
    :exclude-members: forward
    :members: