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

Lysandre's avatar
Lysandre committed
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
Overview
~~~~~~~~~~~~~~~~~~~~~

The BERT model was proposed in `BERT: Pre-training of Deep Bidirectional Transformers for Language Understanding <https://arxiv.org/abs/1810.04805>`__
by Jacob Devlin, Ming-Wei Chang, Kenton Lee and Kristina Toutanova. It's a bidirectional transformer
pre-trained using a combination of masked language modeling objective and next sentence prediction
on a large corpus comprising the Toronto Book Corpus and Wikipedia.

The abstract from the paper is the following:

*We introduce a new language representation model called BERT, which stands for Bidirectional Encoder Representations
from Transformers. Unlike recent language representation models, BERT is designed to pre-train deep bidirectional
representations from unlabeled text by jointly conditioning on both left and right context in all layers. As a result,
the pre-trained BERT model can be fine-tuned with just one additional output layer to create state-of-the-art models
for a wide range of tasks, such as question answering and language inference, without substantial task-specific
architecture modifications.*

*BERT is conceptually simple and empirically powerful. It obtains new state-of-the-art results on eleven natural
language processing tasks, including pushing the GLUE score to 80.5% (7.7% point absolute improvement), MultiNLI
accuracy to 86.7% (4.6% absolute improvement), SQuAD v1.1 question answering Test F1 to 93.2 (1.5 point absolute
improvement) and SQuAD v2.0 Test F1 to 83.1 (5.1 point absolute improvement).*

Tips:

- BERT is a model with absolute position embeddings so it's usually advised to pad the inputs on
  the right rather than the left.


BertConfig
33
34
~~~~~~~~~~~~~~~~~~~~~

35
.. autoclass:: transformers.BertConfig
36
    :members:
37
38


Lysandre's avatar
Lysandre committed
39
BertTokenizer
40
~~~~~~~~~~~~~~~~~~~~~
41

42
.. autoclass:: transformers.BertTokenizer
43
    :members:
44
45


Lysandre's avatar
Lysandre committed
46
BertModel
47
48
~~~~~~~~~~~~~~~~~~~~

49
.. autoclass:: transformers.BertModel
50
51
52
    :members:


Lysandre's avatar
Lysandre committed
53
BertForPreTraining
54
55
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

56
.. autoclass:: transformers.BertForPreTraining
57
58
59
    :members:


Lysandre's avatar
Lysandre committed
60
BertForMaskedLM
61
62
~~~~~~~~~~~~~~~~~~~~~~~~~~

63
.. autoclass:: transformers.BertForMaskedLM
64
65
66
    :members:


Lysandre's avatar
Lysandre committed
67
BertForNextSentencePrediction
68
69
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

70
.. autoclass:: transformers.BertForNextSentencePrediction
71
72
73
    :members:


Lysandre's avatar
Lysandre committed
74
BertForSequenceClassification
75
76
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

77
.. autoclass:: transformers.BertForSequenceClassification
78
79
80
    :members:


Lysandre's avatar
Lysandre committed
81
BertForMultipleChoice
82
83
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

84
.. autoclass:: transformers.BertForMultipleChoice
85
86
87
    :members:


Lysandre's avatar
Lysandre committed
88
BertForTokenClassification
89
90
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

91
.. autoclass:: transformers.BertForTokenClassification
92
93
94
    :members:


Lysandre's avatar
Lysandre committed
95
BertForQuestionAnswering
96
97
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

98
.. autoclass:: transformers.BertForQuestionAnswering
99
100
    :members:

LysandreJik's avatar
LysandreJik committed
101

Lysandre's avatar
Lysandre committed
102
TFBertModel
LysandreJik's avatar
LysandreJik committed
103
104
~~~~~~~~~~~~~~~~~~~~

105
.. autoclass:: transformers.TFBertModel
LysandreJik's avatar
LysandreJik committed
106
107
108
    :members:


Lysandre's avatar
Lysandre committed
109
TFBertForPreTraining
LysandreJik's avatar
LysandreJik committed
110
111
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

112
.. autoclass:: transformers.TFBertForPreTraining
LysandreJik's avatar
LysandreJik committed
113
114
115
    :members:


Lysandre's avatar
Lysandre committed
116
TFBertForMaskedLM
LysandreJik's avatar
LysandreJik committed
117
118
~~~~~~~~~~~~~~~~~~~~~~~~~~

119
.. autoclass:: transformers.TFBertForMaskedLM
LysandreJik's avatar
LysandreJik committed
120
121
122
    :members:


Lysandre's avatar
Lysandre committed
123
TFBertForNextSentencePrediction
LysandreJik's avatar
LysandreJik committed
124
125
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

126
.. autoclass:: transformers.TFBertForNextSentencePrediction
LysandreJik's avatar
LysandreJik committed
127
128
129
    :members:


Lysandre's avatar
Lysandre committed
130
TFBertForSequenceClassification
LysandreJik's avatar
LysandreJik committed
131
132
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

133
.. autoclass:: transformers.TFBertForSequenceClassification
LysandreJik's avatar
LysandreJik committed
134
135
136
    :members:


Lysandre's avatar
Lysandre committed
137
TFBertForMultipleChoice
LysandreJik's avatar
LysandreJik committed
138
139
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

140
.. autoclass:: transformers.TFBertForMultipleChoice
LysandreJik's avatar
LysandreJik committed
141
142
143
    :members:


Lysandre's avatar
Lysandre committed
144
TFBertForTokenClassification
LysandreJik's avatar
LysandreJik committed
145
146
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

147
.. autoclass:: transformers.TFBertForTokenClassification
LysandreJik's avatar
LysandreJik committed
148
149
150
    :members:


Lysandre's avatar
Lysandre committed
151
TFBertForQuestionAnswering
LysandreJik's avatar
LysandreJik committed
152
153
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

154
.. autoclass:: transformers.TFBertForQuestionAnswering
LysandreJik's avatar
LysandreJik committed
155
156
    :members: