Text Classification
Transformers
PyTorch
TensorBoard
Safetensors
bert
Generated from Trainer
text-embeddings-inference
Instructions to use SADAF-IMAMU/Type_of_relation with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use SADAF-IMAMU/Type_of_relation with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-classification", model="SADAF-IMAMU/Type_of_relation")# pip install -U transformers accelerate # Load model directly from transformers import AutoTokenizer, AutoModelForSequenceClassification tokenizer = AutoTokenizer.from_pretrained("SADAF-IMAMU/Type_of_relation") model = AutoModelForSequenceClassification.from_pretrained("SADAF-IMAMU/Type_of_relation", device_map="auto") - Notebooks
- Google Colab
- Kaggle
Commit ·
fcf5622
1
Parent(s): fc62022
End of training
Browse files- README.md +32 -0
- model.safetensors +1 -1
README.md
CHANGED
|
@@ -2,6 +2,10 @@
|
|
| 2 |
base_model: aubmindlab/bert-base-arabertv02
|
| 3 |
tags:
|
| 4 |
- generated_from_trainer
|
|
|
|
|
|
|
|
|
|
|
|
|
| 5 |
model-index:
|
| 6 |
- name: Type_of_relation
|
| 7 |
results: []
|
|
@@ -13,6 +17,13 @@ should probably proofread and complete it, then remove this comment. -->
|
|
| 13 |
# Type_of_relation
|
| 14 |
|
| 15 |
This model is a fine-tuned version of [aubmindlab/bert-base-arabertv02](https://huggingface.co/aubmindlab/bert-base-arabertv02) on an unknown dataset.
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 16 |
|
| 17 |
## Model description
|
| 18 |
|
|
@@ -41,6 +52,27 @@ The following hyperparameters were used during training:
|
|
| 41 |
- lr_scheduler_type: linear
|
| 42 |
- num_epochs: 15
|
| 43 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 44 |
### Framework versions
|
| 45 |
|
| 46 |
- Transformers 4.36.0
|
|
|
|
| 2 |
base_model: aubmindlab/bert-base-arabertv02
|
| 3 |
tags:
|
| 4 |
- generated_from_trainer
|
| 5 |
+
metrics:
|
| 6 |
+
- precision
|
| 7 |
+
- recall
|
| 8 |
+
- accuracy
|
| 9 |
model-index:
|
| 10 |
- name: Type_of_relation
|
| 11 |
results: []
|
|
|
|
| 17 |
# Type_of_relation
|
| 18 |
|
| 19 |
This model is a fine-tuned version of [aubmindlab/bert-base-arabertv02](https://huggingface.co/aubmindlab/bert-base-arabertv02) on an unknown dataset.
|
| 20 |
+
It achieves the following results on the evaluation set:
|
| 21 |
+
- Loss: 1.3454
|
| 22 |
+
- Macro F1: 0.7875
|
| 23 |
+
- Precision: 0.7834
|
| 24 |
+
- Recall: 0.7949
|
| 25 |
+
- Kappa: 0.6913
|
| 26 |
+
- Accuracy: 0.7949
|
| 27 |
|
| 28 |
## Model description
|
| 29 |
|
|
|
|
| 52 |
- lr_scheduler_type: linear
|
| 53 |
- num_epochs: 15
|
| 54 |
|
| 55 |
+
### Training results
|
| 56 |
+
|
| 57 |
+
| Training Loss | Epoch | Step | Validation Loss | Macro F1 | Precision | Recall | Kappa | Accuracy |
|
| 58 |
+
|:-------------:|:-----:|:-----:|:---------------:|:--------:|:---------:|:------:|:------:|:--------:|
|
| 59 |
+
| 1.1521 | 1.0 | 697 | 0.7730 | 0.7668 | 0.7492 | 0.7976 | 0.6831 | 0.7976 |
|
| 60 |
+
| 0.7985 | 2.0 | 1395 | 0.7075 | 0.7817 | 0.7674 | 0.8003 | 0.6965 | 0.8003 |
|
| 61 |
+
| 0.6333 | 3.0 | 2092 | 0.7101 | 0.7840 | 0.7793 | 0.8078 | 0.7023 | 0.8078 |
|
| 62 |
+
| 0.5307 | 4.0 | 2790 | 0.7471 | 0.7797 | 0.7779 | 0.7989 | 0.6929 | 0.7989 |
|
| 63 |
+
| 0.447 | 5.0 | 3487 | 0.7967 | 0.7826 | 0.7765 | 0.7951 | 0.6916 | 0.7951 |
|
| 64 |
+
| 0.304 | 6.0 | 4185 | 0.8912 | 0.7884 | 0.7836 | 0.7976 | 0.6961 | 0.7976 |
|
| 65 |
+
| 0.2597 | 7.0 | 4882 | 0.9286 | 0.7872 | 0.7820 | 0.7962 | 0.6925 | 0.7962 |
|
| 66 |
+
| 0.1859 | 8.0 | 5580 | 1.0321 | 0.7887 | 0.7845 | 0.7996 | 0.6963 | 0.7996 |
|
| 67 |
+
| 0.1542 | 9.0 | 6277 | 1.0918 | 0.7840 | 0.7801 | 0.7926 | 0.6879 | 0.7926 |
|
| 68 |
+
| 0.135 | 10.0 | 6975 | 1.1611 | 0.7884 | 0.7825 | 0.8035 | 0.6988 | 0.8035 |
|
| 69 |
+
| 0.0894 | 11.0 | 7672 | 1.2353 | 0.7866 | 0.7862 | 0.7911 | 0.6871 | 0.7911 |
|
| 70 |
+
| 0.084 | 12.0 | 8370 | 1.2618 | 0.7875 | 0.7832 | 0.7965 | 0.6920 | 0.7965 |
|
| 71 |
+
| 0.0595 | 13.0 | 9067 | 1.3147 | 0.7847 | 0.7836 | 0.7879 | 0.6844 | 0.7879 |
|
| 72 |
+
| 0.0472 | 14.0 | 9765 | 1.3424 | 0.7872 | 0.7839 | 0.7942 | 0.6897 | 0.7942 |
|
| 73 |
+
| 0.0422 | 14.99 | 10455 | 1.3454 | 0.7875 | 0.7834 | 0.7949 | 0.6913 | 0.7949 |
|
| 74 |
+
|
| 75 |
+
|
| 76 |
### Framework versions
|
| 77 |
|
| 78 |
- Transformers 4.36.0
|
model.safetensors
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 540849212
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:2ffeb432144d9a24a464c03594a83760b71e01e221a67f9f93a10c3c7c82aea9
|
| 3 |
size 540849212
|