flaubert.rst 5.2 KB
Newer Older
Lysandre's avatar
Lysandre committed
1
FlauBERT
Sylvain Gugger's avatar
Sylvain Gugger committed
2
-----------------------------------------------------------------------------------------------------------------------
Lysandre's avatar
Lysandre committed
3

Sylvain Gugger's avatar
Sylvain Gugger committed
4
Overview
Sylvain Gugger's avatar
Sylvain Gugger committed
5
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
Sylvain Gugger's avatar
Sylvain Gugger committed
6

Sylvain Gugger's avatar
Sylvain Gugger committed
7
8
9
The FlauBERT model was proposed in the paper `FlauBERT: Unsupervised Language Model Pre-training for French
<https://arxiv.org/abs/1912.05372>`__ by Hang Le et al. It's a transformer model pretrained using a masked language
modeling (MLM) objective (like BERT).
Lysandre's avatar
Lysandre committed
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25

The abstract from the paper is the following:

*Language models have become a key step to achieve state-of-the art results in many different Natural Language
Processing (NLP) tasks. Leveraging the huge amount of unlabeled texts nowadays available, they provide an efficient
way to pre-train continuous word representations that can be fine-tuned for a downstream task, along with their
contextualization at the sentence level. This has been widely demonstrated for English using contextualized
representations (Dai and Le, 2015; Peters et al., 2018; Howard and Ruder, 2018; Radford et al., 2018; Devlin et
al., 2019; Yang et al., 2019b). In this paper, we introduce and share FlauBERT, a model learned on a very large
and heterogeneous French corpus. Models of different sizes are trained using the new CNRS (French National Centre
for Scientific Research) Jean Zay supercomputer. We apply our French language models to diverse NLP tasks (text
classification, paraphrasing, natural language inference, parsing, word sense disambiguation) and show that most
of the time they outperform other pre-training approaches. Different versions of FlauBERT as well as a unified
evaluation protocol for the downstream tasks, called FLUE (French Language Understanding Evaluation), are shared
to the research community for further reproducible experiments in French NLP.*

Sylvain Gugger's avatar
Sylvain Gugger committed
26
The original code can be found `here <https://github.com/getalp/Flaubert>`__.
27

Lysandre's avatar
Lysandre committed
28
29

FlaubertConfig
Sylvain Gugger's avatar
Sylvain Gugger committed
30
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
Lysandre's avatar
Lysandre committed
31
32
33
34
35
36

.. autoclass:: transformers.FlaubertConfig
    :members:


FlaubertTokenizer
Sylvain Gugger's avatar
Sylvain Gugger committed
37
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
Lysandre's avatar
Lysandre committed
38
39
40
41
42
43

.. autoclass:: transformers.FlaubertTokenizer
    :members:


FlaubertModel
Sylvain Gugger's avatar
Sylvain Gugger committed
44
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
Lysandre's avatar
Lysandre committed
45
46

.. autoclass:: transformers.FlaubertModel
Sylvain Gugger's avatar
Sylvain Gugger committed
47
    :members: forward
Lysandre's avatar
Lysandre committed
48
49
50


FlaubertWithLMHeadModel
Sylvain Gugger's avatar
Sylvain Gugger committed
51
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
Lysandre's avatar
Lysandre committed
52
53

.. autoclass:: transformers.FlaubertWithLMHeadModel
Sylvain Gugger's avatar
Sylvain Gugger committed
54
    :members: forward
Lysandre's avatar
Lysandre committed
55
56
57


FlaubertForSequenceClassification
Sylvain Gugger's avatar
Sylvain Gugger committed
58
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
Lysandre's avatar
Lysandre committed
59
60

.. autoclass:: transformers.FlaubertForSequenceClassification
Sylvain Gugger's avatar
Sylvain Gugger committed
61
    :members: forward
Lysandre's avatar
Lysandre committed
62
63


64
FlaubertForMultipleChoice
Sylvain Gugger's avatar
Sylvain Gugger committed
65
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
66
67

.. autoclass:: transformers.FlaubertForMultipleChoice
Sylvain Gugger's avatar
Sylvain Gugger committed
68
    :members: forward
69
70


71
FlaubertForTokenClassification
Sylvain Gugger's avatar
Sylvain Gugger committed
72
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
73
74

.. autoclass:: transformers.FlaubertForTokenClassification
Sylvain Gugger's avatar
Sylvain Gugger committed
75
    :members: forward
76
77


Lysandre's avatar
Lysandre committed
78
FlaubertForQuestionAnsweringSimple
Sylvain Gugger's avatar
Sylvain Gugger committed
79
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
Lysandre's avatar
Lysandre committed
80
81

.. autoclass:: transformers.FlaubertForQuestionAnsweringSimple
Sylvain Gugger's avatar
Sylvain Gugger committed
82
    :members: forward
Lysandre's avatar
Lysandre committed
83
84
85


FlaubertForQuestionAnswering
Sylvain Gugger's avatar
Sylvain Gugger committed
86
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
Lysandre's avatar
Lysandre committed
87
88

.. autoclass:: transformers.FlaubertForQuestionAnswering
Sylvain Gugger's avatar
Sylvain Gugger committed
89
    :members: forward
Lysandre's avatar
Lysandre committed
90
91


Sylvain Gugger's avatar
Sylvain Gugger committed
92
TFFlaubertModel
Sylvain Gugger's avatar
Sylvain Gugger committed
93
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
Sylvain Gugger's avatar
Sylvain Gugger committed
94
95

.. autoclass:: transformers.TFFlaubertModel
Sylvain Gugger's avatar
Sylvain Gugger committed
96
    :members: call
Sylvain Gugger's avatar
Sylvain Gugger committed
97
98
99


TFFlaubertWithLMHeadModel
Sylvain Gugger's avatar
Sylvain Gugger committed
100
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
Sylvain Gugger's avatar
Sylvain Gugger committed
101
102

.. autoclass:: transformers.TFFlaubertWithLMHeadModel
Sylvain Gugger's avatar
Sylvain Gugger committed
103
    :members: call
Sylvain Gugger's avatar
Sylvain Gugger committed
104
105
106


TFFlaubertForSequenceClassification
Sylvain Gugger's avatar
Sylvain Gugger committed
107
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
Sylvain Gugger's avatar
Sylvain Gugger committed
108
109

.. autoclass:: transformers.TFFlaubertForSequenceClassification
Sylvain Gugger's avatar
Sylvain Gugger committed
110
    :members: call
Sylvain Gugger's avatar
Sylvain Gugger committed
111
112
113


TFFlaubertForMultipleChoice
Sylvain Gugger's avatar
Sylvain Gugger committed
114
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
Sylvain Gugger's avatar
Sylvain Gugger committed
115
116

.. autoclass:: transformers.TFFlaubertForMultipleChoice
Sylvain Gugger's avatar
Sylvain Gugger committed
117
    :members: call
Sylvain Gugger's avatar
Sylvain Gugger committed
118
119
120


TFFlaubertForTokenClassification
Sylvain Gugger's avatar
Sylvain Gugger committed
121
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
Sylvain Gugger's avatar
Sylvain Gugger committed
122
123

.. autoclass:: transformers.TFFlaubertForTokenClassification
Sylvain Gugger's avatar
Sylvain Gugger committed
124
    :members: call
Sylvain Gugger's avatar
Sylvain Gugger committed
125
126
127


TFFlaubertForQuestionAnsweringSimple
Sylvain Gugger's avatar
Sylvain Gugger committed
128
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
Sylvain Gugger's avatar
Sylvain Gugger committed
129
130

.. autoclass:: transformers.TFFlaubertForQuestionAnsweringSimple
Sylvain Gugger's avatar
Sylvain Gugger committed
131
    :members: call