mirror of
https://github.com/storytold/storyteller-ml.git
synced 2026-10-13 23:53:02 +00:00
Brandon Thomas
74 lines
2.4 KiB
ReStructuredText
Executable File
74 lines
2.4 KiB
ReStructuredText
Executable File
NeMo NLP collection API
|
|
=======================
|
|
|
|
Model Classes
|
|
-------------
|
|
|
|
.. autoclass:: nemo.collections.nlp.models.TextClassificationModel
|
|
:show-inheritance:
|
|
:members: setup_training_data, setup_optimization, setup_validation_data, setup_test_data, register_artifact, classifytext
|
|
|
|
.. autoclass:: nemo.collections.nlp.models.GLUEModel
|
|
:show-inheritance:
|
|
:members: setup_training_data, setup_optimization, setup_validation_data, setup_test_data, register_artifact
|
|
|
|
.. autoclass:: nemo.collections.nlp.models.PunctuationCapitalizationModel
|
|
:show-inheritance:
|
|
:members: add_punctuation_capitalization, setup_training_data, setup_optimization, setup_validation_data, setup_test_data, multi_validation_epoch_end, register_artifact
|
|
|
|
.. autoclass:: nemo.collections.nlp.models.TokenClassificationModel
|
|
:show-inheritance:
|
|
:members: setup_training_data, setup_optimization, setup_validation_data, setup_test_data, register_artifact
|
|
|
|
.. autoclass:: nemo.collections.nlp.models.QAModel
|
|
:show-inheritance:
|
|
:members: setup_training_data, setup_optimization, setup_validation_data, setup_test_data, inference, validation_epoch_end, test_epoch_end
|
|
|
|
.. autoclass:: nemo.collections.nlp.models.BERTLMModel
|
|
:show-inheritance:
|
|
:members: setup_training_data, setup_optimization
|
|
|
|
Modules
|
|
-------
|
|
|
|
.. autoclass:: nemo.collections.nlp.modules.BertModule
|
|
:show-inheritance:
|
|
:members:
|
|
|
|
.. autoclass:: nemo.collections.nlp.modules.MegatronBertEncoder
|
|
:show-inheritance:
|
|
:members:
|
|
|
|
.. autoclass:: nemo.collections.nlp.modules.AlbertEncoder
|
|
:show-inheritance:
|
|
:members:
|
|
|
|
.. autoclass:: nemo.collections.nlp.modules.BertEncoder
|
|
:show-inheritance:
|
|
:members:
|
|
|
|
.. autoclass:: nemo.collections.nlp.modules.DistilBertEncoder
|
|
:show-inheritance:
|
|
:members:
|
|
|
|
.. autoclass:: nemo.collections.nlp.modules.RobertaEncoder
|
|
:show-inheritance:
|
|
:members:
|
|
|
|
.. autoclass:: nemo.collections.nlp.modules.SequenceClassifier
|
|
:show-inheritance:
|
|
:members:
|
|
|
|
.. autoclass:: nemo.collections.nlp.modules.SequenceRegression
|
|
:show-inheritance:
|
|
:members:
|
|
|
|
.. autoclass:: nemo.collections.nlp.modules.SequenceTokenClassifier
|
|
:show-inheritance:
|
|
:members:
|
|
|
|
.. autofunction:: nemo.collections.nlp.modules.get_lm_model
|
|
|
|
.. autofunction:: nemo.collections.nlp.modules.get_pretrained_lm_models_list
|
|
|
|
.. autofunction:: nemo.collections.nlp.modules.get_megatron_lm_models_list |