flaubert.rst 5.88 KB
Newer Older
Sylvain Gugger's avatar
Sylvain Gugger committed
1
2
3
4
5
6
7
8
9
10
11
12
.. 
    Copyright 2020 The HuggingFace Team. All rights reserved.

    Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with
    the License. You may obtain a copy of the License at

        http://www.apache.org/licenses/LICENSE-2.0

    Unless required by applicable law or agreed to in writing, software distributed under the License is distributed on
    an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the License for the
    specific language governing permissions and limitations under the License.

Lysandre's avatar
Lysandre committed
13
FlauBERT
Sylvain Gugger's avatar
Sylvain Gugger committed
14
-----------------------------------------------------------------------------------------------------------------------
Lysandre's avatar
Lysandre committed
15

Sylvain Gugger's avatar
Sylvain Gugger committed
16
Overview
Sylvain Gugger's avatar
Sylvain Gugger committed
17
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
Sylvain Gugger's avatar
Sylvain Gugger committed
18

Sylvain Gugger's avatar
Sylvain Gugger committed
19
20
21
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
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
Sylvain Gugger's avatar
Sylvain Gugger committed
26
27
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
Lysandre's avatar
Lysandre committed
28
contextualization at the sentence level. This has been widely demonstrated for English using contextualized
Sylvain Gugger's avatar
Sylvain Gugger committed
29
30
31
32
33
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
34
time they outperform other pretraining approaches. Different versions of FlauBERT as well as a unified evaluation
Sylvain Gugger's avatar
Sylvain Gugger committed
35
36
protocol for the downstream tasks, called FLUE (French Language Understanding Evaluation), are shared to the research
community for further reproducible experiments in French NLP.*
Lysandre's avatar
Lysandre committed
37

38
39
This model was contributed by `formiel <https://huggingface.co/formiel>`__. The original code can be found `here
<https://github.com/getalp/Flaubert>`__.
40

Lysandre's avatar
Lysandre committed
41
42

FlaubertConfig
Sylvain Gugger's avatar
Sylvain Gugger committed
43
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
Lysandre's avatar
Lysandre committed
44
45
46
47
48
49

.. autoclass:: transformers.FlaubertConfig
    :members:


FlaubertTokenizer
Sylvain Gugger's avatar
Sylvain Gugger committed
50
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
Lysandre's avatar
Lysandre committed
51
52
53
54
55
56

.. autoclass:: transformers.FlaubertTokenizer
    :members:


FlaubertModel
Sylvain Gugger's avatar
Sylvain Gugger committed
57
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
Lysandre's avatar
Lysandre committed
58
59

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


FlaubertWithLMHeadModel
Sylvain Gugger's avatar
Sylvain Gugger committed
64
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
Lysandre's avatar
Lysandre committed
65
66

.. autoclass:: transformers.FlaubertWithLMHeadModel
Sylvain Gugger's avatar
Sylvain Gugger committed
67
    :members: forward
Lysandre's avatar
Lysandre committed
68
69
70


FlaubertForSequenceClassification
Sylvain Gugger's avatar
Sylvain Gugger committed
71
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
Lysandre's avatar
Lysandre committed
72
73

.. autoclass:: transformers.FlaubertForSequenceClassification
Sylvain Gugger's avatar
Sylvain Gugger committed
74
    :members: forward
Lysandre's avatar
Lysandre committed
75
76


77
FlaubertForMultipleChoice
Sylvain Gugger's avatar
Sylvain Gugger committed
78
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
79
80

.. autoclass:: transformers.FlaubertForMultipleChoice
Sylvain Gugger's avatar
Sylvain Gugger committed
81
    :members: forward
82
83


84
FlaubertForTokenClassification
Sylvain Gugger's avatar
Sylvain Gugger committed
85
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
86
87

.. autoclass:: transformers.FlaubertForTokenClassification
Sylvain Gugger's avatar
Sylvain Gugger committed
88
    :members: forward
89
90


Lysandre's avatar
Lysandre committed
91
FlaubertForQuestionAnsweringSimple
Sylvain Gugger's avatar
Sylvain Gugger committed
92
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
Lysandre's avatar
Lysandre committed
93
94

.. autoclass:: transformers.FlaubertForQuestionAnsweringSimple
Sylvain Gugger's avatar
Sylvain Gugger committed
95
    :members: forward
Lysandre's avatar
Lysandre committed
96
97
98


FlaubertForQuestionAnswering
Sylvain Gugger's avatar
Sylvain Gugger committed
99
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
Lysandre's avatar
Lysandre committed
100
101

.. autoclass:: transformers.FlaubertForQuestionAnswering
Sylvain Gugger's avatar
Sylvain Gugger committed
102
    :members: forward
Lysandre's avatar
Lysandre committed
103
104


Sylvain Gugger's avatar
Sylvain Gugger committed
105
TFFlaubertModel
Sylvain Gugger's avatar
Sylvain Gugger committed
106
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
Sylvain Gugger's avatar
Sylvain Gugger committed
107
108

.. autoclass:: transformers.TFFlaubertModel
Sylvain Gugger's avatar
Sylvain Gugger committed
109
    :members: call
Sylvain Gugger's avatar
Sylvain Gugger committed
110
111
112


TFFlaubertWithLMHeadModel
Sylvain Gugger's avatar
Sylvain Gugger committed
113
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
Sylvain Gugger's avatar
Sylvain Gugger committed
114
115

.. autoclass:: transformers.TFFlaubertWithLMHeadModel
Sylvain Gugger's avatar
Sylvain Gugger committed
116
    :members: call
Sylvain Gugger's avatar
Sylvain Gugger committed
117
118
119


TFFlaubertForSequenceClassification
Sylvain Gugger's avatar
Sylvain Gugger committed
120
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
Sylvain Gugger's avatar
Sylvain Gugger committed
121
122

.. autoclass:: transformers.TFFlaubertForSequenceClassification
Sylvain Gugger's avatar
Sylvain Gugger committed
123
    :members: call
Sylvain Gugger's avatar
Sylvain Gugger committed
124
125
126


TFFlaubertForMultipleChoice
Sylvain Gugger's avatar
Sylvain Gugger committed
127
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
Sylvain Gugger's avatar
Sylvain Gugger committed
128
129

.. autoclass:: transformers.TFFlaubertForMultipleChoice
Sylvain Gugger's avatar
Sylvain Gugger committed
130
    :members: call
Sylvain Gugger's avatar
Sylvain Gugger committed
131
132
133


TFFlaubertForTokenClassification
Sylvain Gugger's avatar
Sylvain Gugger committed
134
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
Sylvain Gugger's avatar
Sylvain Gugger committed
135
136

.. autoclass:: transformers.TFFlaubertForTokenClassification
Sylvain Gugger's avatar
Sylvain Gugger committed
137
    :members: call
Sylvain Gugger's avatar
Sylvain Gugger committed
138
139
140


TFFlaubertForQuestionAnsweringSimple
Sylvain Gugger's avatar
Sylvain Gugger committed
141
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
Sylvain Gugger's avatar
Sylvain Gugger committed
142
143

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