File size: 1,246 Bytes
f024a65
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
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
33
34
35
36
# FlauBert finetuned on French cooking recipes

This model is finetuned on a sequence classification task that associate each sequence to the appropriate recipe category. 

### How to use it ?

```python
from transformers import AutoTokenizer, AutoModelForSequenceClassification
from transformers import TextClassificationPipeline

loaded_tokenizer = AutoTokenizer.from_pretrained("nbouali/flaubert-base-uncased-finetuned-cooking")
loaded_model = AutoModelForSequenceClassification.from_pretrained("nbouali/flaubert-base-uncased-finetuned-cooking")

nlp = TextClassificationPipeline(model=loaded_model,tokenizer=loaded_tokenizer,task="Recipe classification")

print(nlp("Lasagnes à la bolognaise"))
```

```
[{'label': 'LABEL_6', 'score': 0.9921900033950806}]
```
### Label encoding 

| label | Recipe Category |
|:------:|:--------------:|
|     0 |'Accompagnement' |
|     1 |  'Amuse-gueule' |
|     2 | 'Boisson' |
|     3 | 'Confiserie' |
|     4 | 'Dessert'|
|     5 | 'Entrée' |
|     6 |'Plat principal' |
|     7 | 'Sauce' |


If you would like to know more about this model you can refer to [our blog post](https://medium.com/unify-data-office/a-cooking-language-model-fine-tuned-on-dozens-of-thousands-of-french-recipes-bcdb8e560571)