easynlp.modelzoo
===============================


BERT
-----------------------------------
.. autoclass:: easynlp.modelzoo.models.bert.modeling_bert.BertConfig
    :members:
    :undoc-members:
    :exclude-members:

.. autoclass:: easynlp.modelzoo.models.bert.modeling_bert.BertForPreTrainingOutput
    :members:
    :undoc-members:
    :exclude-members: pretrained_model_archive_map, pretrained_config_archive_map

.. autoclass:: easynlp.modelzoo.models.bert.modeling_bert.BertModel
    :members:
    :undoc-members:
    :exclude-members: pretrained_model_archive_map, pretrained_config_archive_map

    :members:
    :undoc-members:
    :exclude-members: pretrained_model_archive_map, pretrained_config_archive_map

.. autoclass:: easynlp.modelzoo.models.bert.modeling_bert.BertLMHeadModel
    :members:
    :undoc-members:
    :exclude-members: pretrained_model_archive_map, pretrained_config_archive_map

.. autoclass:: easynlp.modelzoo.models.bert.modeling_bert.BertForNextSentencePrediction
    :members:
    :undoc-members:
    :exclude-members: pretrained_model_archive_map, pretrained_config_archive_map


TextCNN
-----------------------------------
.. autoclass:: easynlp.modelzoo.models.cnn.TextCNNConfig
    :members:
    :undoc-members:
    :exclude-members: from_dict

.. autoclass:: easynlp.modelzoo.models.cnn.TextCNNEncoder
    :members:
    :undoc-members:
    :exclude-members:


DKPLM
-----------------------------------

.. autoclass:: easynlp.modelzoo.models.dkplm.modeling_dkplm.DkplmConfig
    :members:
    :undoc-members:
    :exclude-members:

.. autoclass:: easynlp.modelzoo.models.dkplm.modeling_dkplm.DkplmForPreTrainingOutput
    :members:
    :undoc-members:
    :exclude-members: pretrained_model_archive_map, pretrained_config_archive_map

.. autoclass:: easynlp.modelzoo.models.dkplm.modeling_dkplm.DkplmModel
    :members:
    :undoc-members:
    :exclude-members: pretrained_model_archive_map, pretrained_config_archive_map

    :members:
    :undoc-members:
    :exclude-members: pretrained_model_archive_map, pretrained_config_archive_map

.. autoclass:: easynlp.modelzoo.models.dkplm.modeling_dkplm.DkplmLMHeadModel
    :members:
    :undoc-members:
    :exclude-members: pretrained_model_archive_map, pretrained_config_archive_map

.. autoclass:: easynlp.modelzoo.models.dkplm.modeling_dkplm.DkplmForNextSentencePrediction
    :members:
    :undoc-members:
    :exclude-members: pretrained_model_archive_map, pretrained_config_archive_map

GEEP
-----------------------------------

.. autoclass:: easynlp.modelzoo.models.geep.modeling_geep.GEEPModel
    :members:
    :undoc-members:
    :exclude-members:

RoBERTa
-----------------------------------
.. autoclass:: easynlp.modelzoo.models.roberta.modeling_roberta.RobertaConfig
    :members:
    :undoc-members:
    :exclude-members:

.. autoclass:: easynlp.modelzoo.models.roberta.modeling_roberta.RobertaModel
    :members:
    :undoc-members:
    :exclude-members: pretrained_model_archive_map, pretrained_config_archive_map

.. autoclass:: easynlp.modelzoo.models.roberta.modeling_roberta.RobertaForCausalLM
    :members:
    :undoc-members:
    :exclude-members: pretrained_model_archive_map, pretrained_config_archive_map

GPT2
-----------------------------------
.. autoclass:: easynlp.modelzoo.models.gpt2.modeling_gpt2.GPT2Config
    :members:
    :undoc-members:
    :exclude-members:

.. autoclass:: easynlp.modelzoo.models.gpt2.modeling_gpt2.GPT2DoubleHeadsModelOutput
    :members:
    :undoc-members:
    :exclude-members: pretrained_model_archive_map, pretrained_config_archive_map

.. autoclass:: easynlp.modelzoo.models.gpt2.modeling_gpt2.GPT2DoubleHeadsModel
    :members:
    :undoc-members:
    :exclude-members: pretrained_model_archive_map, pretrained_config_archive_map
