logitorch.models.lreasoner
==========================

.. py:module:: logitorch.models.lreasoner


Classes
-------

.. autoapisummary::

   logitorch.models.lreasoner.LReasoner


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

.. py:class:: LReasoner(pretrained_model: str)

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


   
   Initializes an instance of the LReasoner class.

   Args:
       pretrained_model (str): The name or path of the pre-trained model to use.

   Returns:
       None


   .. py:method:: forward(**x: Dict[str, torch.Tensor]) -> transformers.modeling_outputs.SequenceClassifierOutput

      Performs forward pass through the LReasoner model.

      Args:
          **x (Dict[str, torch.Tensor]): Input tensors for the model.

      Returns:
          SequenceClassifierOutput: The output of the model.



   .. py:method:: predict(**x: Dict[str, torch.Tensor]) -> List[str]

      Predicts the class labels for the input tensors.

      Args:
          **x (Dict[str, torch.Tensor]): Input tensors for prediction.

      Returns:
          List[str]: The predicted class labels.



   .. py:attribute:: model