logitorch.data_collators.bertnot_collator
=========================================

.. py:module:: logitorch.data_collators.bertnot_collator


Classes
-------

.. autoapisummary::

   logitorch.data_collators.bertnot_collator.BERTNOTTextualEntailmentCollator
   logitorch.data_collators.bertnot_collator.BERTNOTWiki20KCollator


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

.. py:class:: BERTNOTTextualEntailmentCollator(pretrained_tokenizer: str)

   Collator for BERTNOTTextualEntailment dataset.

   Args:
       | pretrained_tokenizer (str): Pretrained tokenizer name or path.

   Returns:
       | Tuple[Dict[str, torch.Tensor], torch.Tensor]: Batch of tokenized premises, hypotheses, and labels.


   .. py:attribute:: tokenizer


.. py:class:: BERTNOTWiki20KCollator(pretrained_tokenizer: str)

   Collator for BERTNOTWiki20K dataset.

   Args:
       | pretrained_tokenizer (str): Pretrained tokenizer name or path.

   Returns:
       | Tuple[Dict[str, torch.Tensor], torch.Tensor]: Batch of tokenized sentences and labels.


   .. py:attribute:: tokenizer