File size: 1,094 Bytes
5f3d17d
 
902bfcf
 
 
 
 
5f3d17d
 
902bfcf
5f3d17d
902bfcf
5f3d17d
902bfcf
5f3d17d
902bfcf
5f3d17d
902bfcf
5f3d17d
902bfcf
 
5f3d17d
902bfcf
5f3d17d
902bfcf
 
 
 
5f3d17d
902bfcf
 
 
 
5f3d17d
 
902bfcf
 
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
37
38
---
library_name: transformers
tags:
  - bangla
  - bangla-classifier
  - binary-classifier
  - text-classifier
---

# Bangla Binary Text Classifier

## Description

This is a **Bangla binary sentiment classification** model, fine-tuned on top of [`csebuetnlp/banglabert`](https://huggingface.co/csebuetnlp/banglabert). The model was trained using the [**SayedShaun/sentigold**](https://huggingface.co/datasets/SayedShaun/sentigold)

---

## How to Use

```python
from transformers import pipeline

pipe = pipeline("text-classification", model="SayedShaun/bangla-classifier-binary")

response = pipe("এটা যে এত খারাপ আগে জানতাম না।")
print(response)
# Output: [{'label': 'LABEL_0', 'score': 0.9765}]
```

## Result
| Training Loss | Validation Loss | Accuracy  | Precision | Recall   | F1 Score  |
|---------------|-----------------|-----------|-----------|----------|-----------|
| 0.354600      | 0.396599        | 0.825143  | 0.812587  | 0.842483 | 0.827265  |


# Source Code
Source code can be found in `files and versions` as `finetune.py`