---
license: apache-2.0
language:
- fa
metrics:
- f1
- accuracy
base_model:
- HooshvareLab/bert-fa-base-uncased
library_name: transformers
---

# bert-fa-sentiment-taghche

Fine-tuned Persian sentiment analysis model based on [`HooshvareLab/bert-fa-base-uncased`](https://huggingface.co/HooshvareLab/bert-fa-base-uncased) for binary sentiment classification on the Taaghche Persian review dataset.

## Dataset

The model was trained on the Taaghche Persian reviews dataset available on Kaggle:

* [Taaghche Dataset](https://www.kaggle.com/saeedtqp/taaghche)

## Model Details

* **Base model:** `HooshvareLab/bert-fa-base-uncased`
* **Task:** Sentiment Classification
* **Language:** Persian (Farsi)
* **Framework:** Transformers + PyTorch

## Usage

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

model_name = "aysangh/bert-fa-sentiment-taghche"
model = AutoModelForSequenceClassification.from_pretrained(model_name)
tokenizer = AutoTokenizer.from_pretrained(model_name)

def predict_sentiment(text):
    inputs = tokenizer(text, return_tensors="pt", truncation=True, max_length=256, padding=True)
    with torch.no_grad():
        outputs = model(**inputs)
    probs = torch.softmax(outputs.logits, dim=-1)
    pred = torch.argmax(probs, dim=-1).item()
    confidence = probs[0][pred].item()
    return {
    "label": "positive" if pred == 0 else "negative",
    "confidence": confidence
}

text = "ﺦﯿﻠﯾ ﺏﺩ ﺏﻭﺩ، ﺎﺻﻻ ﭗﯿﺸﻨﻫﺍﺩ ﻦﻤﯾ<200c>ﮑﻨﻣ."

result = predict_sentiment(text)
print(result)

# Output:
# {'label': 'negative', 'confidence': 0.9619705080986023}
```

## Repository

Training and inference scripts are available on [GitHub](https://github.com/aysangh/persian-sentiment-analysis).