Hugging Face's logo

ace-in-the-hole
/
cita_test

Text Classification
Transformers
Safetensors
roberta
Generated from Trainer
Inference Endpoints
Model card Files Community
ace-in-the-hole commited on
Commit
d4f2898
·
verified ·
1 Parent(s): b65f836

End of training

Browse files
Files changed (3) hide show
  1. README.md +14 -14
  2. config.json +1 -0
  3. model.safetensors +1 -1
README.md CHANGED
@@ -21,11 +21,11 @@ should probably proofread and complete it, then remove this comment. -->
21
 
22
  This model is a fine-tuned version of [vinai/phobert-base-v2](https://huggingface.co/vinai/phobert-base-v2) on the None dataset.
23
  It achieves the following results on the evaluation set:
24
- - Loss: 0.4167
25
- - Accuracy: 0.9077
26
- - F1: 0.8903
27
- - Precision: 0.8902
28
- - Recall: 0.8904
29
 
30
  ## Model description
31
 
@@ -58,15 +58,15 @@ The following hyperparameters were used during training:
58
 
59
  | Training Loss | Epoch | Step | Validation Loss | Accuracy | F1 | Precision | Recall |
60
  |:-------------:|:-------:|:----:|:---------------:|:--------:|:------:|:---------:|:------:|
61
- | 0.6924 | 2.1277 | 100 | 0.3109 | 0.8863 | 0.8704 | 0.8590 | 0.8875 |
62
- | 0.3068 | 4.2553 | 200 | 0.2648 | 0.9187 | 0.9031 | 0.9044 | 0.9018 |
63
- | 0.3068 | 6.3830 | 300 | 0.3089 | 0.9173 | 0.9006 | 0.9056 | 0.8961 |
64
- | 0.1813 | 8.5106 | 400 | 0.3284 | 0.921 | 0.9056 | 0.9081 | 0.9031 |
65
- | 0.1813 | 10.6383 | 500 | 0.3317 | 0.905 | 0.8888 | 0.8835 | 0.8948 |
66
- | 0.1185 | 12.7660 | 600 | 0.3676 | 0.896 | 0.8798 | 0.8710 | 0.8912 |
67
- | 0.1185 | 14.8936 | 700 | 0.3970 | 0.9033 | 0.8854 | 0.8846 | 0.8861 |
68
- | 0.079 | 17.0213 | 800 | 0.4143 | 0.9047 | 0.8867 | 0.8867 | 0.8867 |
69
- | 0.079 | 19.1489 | 900 | 0.4167 | 0.9077 | 0.8903 | 0.8902 | 0.8904 |
70
 
71
 
72
  ### Framework versions
 
21
 
22
  This model is a fine-tuned version of [vinai/phobert-base-v2](https://huggingface.co/vinai/phobert-base-v2) on the None dataset.
23
  It achieves the following results on the evaluation set:
24
+ - Loss: 0.3031
25
+ - Accuracy: 0.9183
26
+ - F1: 0.9029
27
+ - Precision: 0.9113
28
+ - Recall: 0.8958
29
 
30
  ## Model description
31
 
 
58
 
59
  | Training Loss | Epoch | Step | Validation Loss | Accuracy | F1 | Precision | Recall |
60
  |:-------------:|:-------:|:----:|:---------------:|:--------:|:------:|:---------:|:------:|
61
+ | 0.6745 | 2.1277 | 100 | 0.2197 | 0.926 | 0.9110 | 0.9256 | 0.8996 |
62
+ | 0.2806 | 4.2553 | 200 | 0.2333 | 0.9293 | 0.9140 | 0.9363 | 0.8979 |
63
+ | 0.2806 | 6.3830 | 300 | 0.2366 | 0.916 | 0.9027 | 0.8998 | 0.9058 |
64
+ | 0.1669 | 8.5106 | 400 | 0.2277 | 0.9283 | 0.9149 | 0.9229 | 0.9080 |
65
+ | 0.1669 | 10.6383 | 500 | 0.2593 | 0.922 | 0.9067 | 0.9183 | 0.8972 |
66
+ | 0.1132 | 12.7660 | 600 | 0.2683 | 0.9243 | 0.9099 | 0.9191 | 0.9022 |
67
+ | 0.1132 | 14.8936 | 700 | 0.2796 | 0.9203 | 0.9051 | 0.9148 | 0.8969 |
68
+ | 0.0818 | 17.0213 | 800 | 0.2948 | 0.9217 | 0.9064 | 0.9175 | 0.8973 |
69
+ | 0.0818 | 19.1489 | 900 | 0.3031 | 0.9183 | 0.9029 | 0.9113 | 0.8958 |
70
 
71
 
72
  ### Framework versions
config.json CHANGED
@@ -19,6 +19,7 @@
19
  "num_hidden_layers": 12,
20
  "pad_token_id": 1,
21
  "position_embedding_type": "absolute",
 
22
  "tokenizer_class": "PhobertTokenizer",
23
  "torch_dtype": "float32",
24
  "transformers_version": "4.45.1",
 
19
  "num_hidden_layers": 12,
20
  "pad_token_id": 1,
21
  "position_embedding_type": "absolute",
22
+ "problem_type": "single_label_classification",
23
  "tokenizer_class": "PhobertTokenizer",
24
  "torch_dtype": "float32",
25
  "transformers_version": "4.45.1",
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:68429966ddd132cb1135f8f25a313ef7a2cd8a70c16af0dae49dfd9970cfbb8f
3
  size 540023384
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9e853f93be48608b14d93bf510dca8e644d8c2f431e4e1afc1af5b796053d5fd
3
  size 540023384