Edit model card

This model is a fine-tuned version of bert-medium-amharic on the amharic-sentiment dataset for sentiment classification. It achieves the following results on the evaluation set:

  • Accuracy: 0.83
  • Precision: 0.84
  • Recall: 0.83
  • F1: 0.83

How to use

You can use this model directly with a pipeline for text classification:

>>> from transformers import pipeline
>>> bert_sentiment = pipeline("text-classification", model="rasyosef/bert-medium-amharic-finetuned-sentiment")
>>> bert_sentiment(["አሪፍ ፊልም ነው።", "ዩክሬን እና ሩስያ ከባድ ውግያ ላይ ናቸው።"])

[{'label': 'positive', 'score': 0.9863048791885376},
 {'label': 'negative', 'score': 0.9570127129554749}]

Code

https://github.com/rasyosef/amharic-sentiment-classification

Downloads last month
14
Safetensors
Model size
40.4M params
Tensor type
F32
·
Inference Examples
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.

Dataset used to train rasyosef/bert-medium-amharic-finetuned-sentiment

Collection including rasyosef/bert-medium-amharic-finetuned-sentiment