DemangeJeremy's picture
Update README.md
1795c95
|
raw
history blame
No virus
1.44 kB
metadata
language: fr
tags:
  - sentiments
  - text-classification
  - flaubert
  - french
  - flaubert-large

Modèle de détection de 4 sentiments avec FlauBERT (mixed, negative, objective, positive)

Les travaux sont actuellement en cours. Je modifierai le modèle ces prochains jours.

Comment l'utiliser ?

from transformers import AutoTokenizer, AutoModelForSequenceClassification
from transformers import pipeline

loaded_tokenizer = AutoTokenizer.from_pretrained('flaubert/flaubert_large_cased')
loaded_model = AutoModelForSequenceClassification.from_pretrained("DemangeJeremy/4-sentiments-with-flaubert")

nlp = pipeline('sentiment-analysis', model=loaded_model, tokenizer=loaded_tokenizer)

print(nlp("Je suis plutôt confiant."))
[{'label': 'OBJECTIVE', 'score': 0.3320835530757904}]

Résultats de l'évaluation du modèle

Epoch Validation Loss Samples Per Second
1 2.219246 49.476000
2 1.883753 47.259000
3 1.747969 44.957000
4 1.695606 43.872000
5 1.641470 45.726000

Citation

Pour toute utilisation de ce modèle, merci d'utiliser cette citation :

Jérémy Demange, Four sentiments with FlauBERT, (2021), Hugging Face repository, https://huggingface.co/DemangeJeremy/4-sentiments-with-flaubert