bert.rst 1.46 KB
Newer Older
1
2
3
BERT
----------------------------------------------------

4
``BertConfig``
5
6
~~~~~~~~~~~~~~~~~~~~~

7
8
.. autoclass:: pytorch_pretrained_bert.BertConfig
    :members:
9
10


11
12
``BertTokenizer``
~~~~~~~~~~~~~~~~~~~~~
13

14
15
.. autoclass:: pytorch_pretrained_bert.BertTokenizer
    :members:
16
17
18
19
20


``BertAdam``
~~~~~~~~~~~~~~~~

21
22
.. autoclass:: pytorch_pretrained_bert.BertAdam
    :members:
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78

1. ``BertModel``
~~~~~~~~~~~~~~~~~~~~

.. autoclass:: pytorch_pretrained_bert.BertModel
    :members:


2. ``BertForPreTraining``
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

.. autoclass:: pytorch_pretrained_bert.BertForPreTraining
    :members:


3. ``BertForMaskedLM``
~~~~~~~~~~~~~~~~~~~~~~~~~~

.. autoclass:: pytorch_pretrained_bert.BertForMaskedLM
    :members:


4. ``BertForNextSentencePrediction``
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

.. autoclass:: pytorch_pretrained_bert.BertForNextSentencePrediction
    :members:


5. ``BertForSequenceClassification``
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

.. autoclass:: pytorch_pretrained_bert.BertForSequenceClassification
    :members:


6. ``BertForMultipleChoice``
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

.. autoclass:: pytorch_pretrained_bert.BertForMultipleChoice
    :members:


7. ``BertForTokenClassification``
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

.. autoclass:: pytorch_pretrained_bert.BertForTokenClassification
    :members:


8. ``BertForQuestionAnswering``
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

.. autoclass:: pytorch_pretrained_bert.BertForQuestionAnswering
    :members: