:py:mod:`ontolearn.nces_trainer`
================================

.. py:module:: ontolearn.nces_trainer

.. autoapi-nested-parse::

   NCES trainer.



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

Classes
~~~~~~~

.. autoapisummary::

   ontolearn.nces_trainer.NCESTrainer



Functions
~~~~~~~~~

.. autoapisummary::

   ontolearn.nces_trainer.before_pad



.. py:function:: before_pad(arg)


.. py:class:: NCESTrainer(nces, epochs=300, learning_rate=0.0001, decay_rate=0, clip_value=5.0, num_workers=8, storage_path='./')


   NCES trainer.

   .. py:method:: compute_accuracy(prediction, target)
      :staticmethod:


   .. py:method:: get_optimizer(synthesizer, optimizer='Adam')


   .. py:method:: show_num_learnable_params()


   .. py:method:: collate_batch(batch)


   .. py:method:: map_to_token(idx_array)


   .. py:method:: train(train_dataloader, save_model=True, optimizer='Adam', record_runtime=True)