:py:mod:`ensemble`
==================

.. py:module:: ensemble


Module Contents
---------------

Classes
~~~~~~~

.. autoapisummary::

   ensemble.Ensemble




.. py:class:: Ensemble(modelA, modelB, modelC=None)


   Bases: :py:obj:`torch.nn.Module`

   ensemble through model averaging.


   .. py:method:: forward_head_batch(*, e1_idx, rel_idx)