Edit model card

camembert-fr-covid-tweet-classification

This model is a fine-tune checkpoint of Yanzhu/bertweetfr-base, fine-tuned on SST-2. This model reaches an accuracy of 66.00% on the dev set.

In this dataset, given a tweet, the goal was to infer the underlying topic of the tweet by choosing from four topics classes:

  • chiffres : this means, the tweet talk about statistics of covid.
  • mesures : this means, the tweet talk about measures take by government of covid
  • opinions : this means, the tweet talk about opinion of people like fake new.
  • symptomes : this means, the tweet talk about symptoms or variant of covid.
  • divers : or other

Pipelining the Model

from transformers import AutoTokenizer, AutoModelForSequenceClassification, pipeline
tokenizer = AutoTokenizer.from_pretrained("Monsia/camembert-fr-covid-tweet-classification")
model = AutoModelForSequenceClassification.from_pretrained("Monsia/camembert-fr-covid-tweet-classification")
nlp_topic_classif = transformers.pipeline('topics-classification', model = model, tokenizer = tokenizer)
nlp_topic_classif("tchai on est morts. on va se faire vacciner et ils vont contrôler comme les marionnettes avec des fils. d'après les '' ont dit ''...")
# Output: [{'label': 'opinions', 'score': 0.831]
Downloads last month
10
This model does not have enough activity to be deployed to Inference API (serverless) yet. Increase its social visibility and check back later, or deploy to Inference Endpoints (dedicated) instead.