Commit
·
b36b492
1
Parent(s):
2935b91
Update README.md
Browse files
README.md
CHANGED
|
@@ -14,7 +14,10 @@ should probably proofread and complete it, then remove this comment. -->
|
|
| 14 |
|
| 15 |
# pft-clf-finetuned
|
| 16 |
|
| 17 |
-
This model is a fine-tuned version of [HooshvareLab/bert-fa-zwnj-base](https://huggingface.co/HooshvareLab/bert-fa-zwnj-base) on an
|
|
|
|
|
|
|
|
|
|
| 18 |
It achieves the following results on the evaluation set:
|
| 19 |
- Loss: 0.0617
|
| 20 |
- Matthews Correlation: 0.9830
|
|
|
|
| 14 |
|
| 15 |
# pft-clf-finetuned
|
| 16 |
|
| 17 |
+
This model is a fine-tuned version of [HooshvareLab/bert-fa-zwnj-base](https://huggingface.co/HooshvareLab/bert-fa-zwnj-base) on an "FarsNews1398" dataset. This dataset contains a collection of news that has been gathered from the farsnews website which is a news agency in Iran. You can download the dataset from [here](https://www.kaggle.com/amirhossein76/farsnews1398). I used category, abstract, and paragraphs of news for doing text classification. "abstract" and "paragraphs" for each news concatenated together and "category" used as a target for classification.
|
| 18 |
+
|
| 19 |
+
The notebook used for fine-tuning can be found [here](https://colab.research.google.com/drive/1jC2dfKRASxCY-b6bJSPkhEJfQkOA30O0?usp=sharing). I've reported loss and Matthews correlation criteria on the validation set.
|
| 20 |
+
|
| 21 |
It achieves the following results on the evaluation set:
|
| 22 |
- Loss: 0.0617
|
| 23 |
- Matthews Correlation: 0.9830
|