Skip to content

Latest commit

 

History

History
41 lines (24 loc) · 1.29 KB

torch_bert.rst

File metadata and controls

41 lines (24 loc) · 1.29 KB
Raw

deeppavlov.models.torch_bert

.. automodule:: deeppavlov.models.torch_bert
   :members:


.. autoclass:: deeppavlov.models.preprocessors.torch_transformers_preprocessor.TorchTransformersPreprocessor

    .. automethod:: __call__


.. autoclass:: deeppavlov.models.preprocessors.torch_transformers_preprocessor.TorchTransformersNerPreprocessor

    .. automethod:: __call__


.. autoclass:: deeppavlov.models.preprocessors.torch_transformers_preprocessor.TorchBertRankerPreprocessor

    .. automethod:: __call__


.. autoclass:: deeppavlov.models.torch_bert.torch_transformers_classifier.TorchTransformersClassifierModel

    .. automethod:: __call__
    .. automethod:: train_on_batch


.. autoclass:: deeppavlov.models.torch_bert.torch_transformers_sequence_tagger.TorchTransformersSequenceTagger

    .. automethod:: __call__
    .. automethod:: train_on_batch


.. autoclass:: deeppavlov.models.torch_bert.torch_transformers_squad.TorchTransformersSquad

    .. automethod:: __call__
    .. automethod:: train_on_batch


.. autoclass:: deeppavlov.models.torch_bert.torch_transformers_squad.TorchTransformersSquadInfer

    .. automethod:: __call__


.. autoclass:: deeppavlov.models.torch_bert.torch_bert_ranker.TorchBertRankerModel

    .. automethod:: __call__
    .. automethod:: train_on_batch