sbaru commited on
Commit
9dd7312
·
verified ·
1 Parent(s): eadd56d

sbaru/news-topic-classifier

Browse files
Files changed (4) hide show
  1. README.md +10 -10
  2. config.json +1 -1
  3. model.safetensors +1 -1
  4. training_args.bin +1 -1
README.md CHANGED
@@ -27,11 +27,11 @@ should probably proofread and complete it, then remove this comment. -->
27
 
28
  This model is a fine-tuned version of [monologg/koelectra-base-v3-discriminator](https://huggingface.co/monologg/koelectra-base-v3-discriminator) on the klue-ynat dataset.
29
  It achieves the following results on the evaluation set:
30
- - Loss: 0.4131
31
- - Accuracy: 0.8589
32
- - Precision: 0.8477
33
- - Recall: 0.8712
34
- - F1: 0.8588
35
 
36
  ## Model description
37
 
@@ -62,14 +62,14 @@ The following hyperparameters were used during training:
62
 
63
  | Training Loss | Epoch | Step | Validation Loss | Accuracy | Precision | Recall | F1 |
64
  |:-------------:|:-----:|:----:|:---------------:|:--------:|:---------:|:------:|:------:|
65
- | 0.3956 | 1.0 | 714 | 0.4286 | 0.8542 | 0.8351 | 0.8768 | 0.8542 |
66
- | 0.2884 | 2.0 | 1428 | 0.4004 | 0.8565 | 0.8476 | 0.8696 | 0.8567 |
67
- | 0.222 | 3.0 | 2142 | 0.4131 | 0.8589 | 0.8477 | 0.8712 | 0.8588 |
68
 
69
 
70
  ### Framework versions
71
 
72
- - Transformers 4.54.1
73
  - Pytorch 2.6.0+cu124
74
  - Datasets 4.0.0
75
- - Tokenizers 0.21.4
 
27
 
28
  This model is a fine-tuned version of [monologg/koelectra-base-v3-discriminator](https://huggingface.co/monologg/koelectra-base-v3-discriminator) on the klue-ynat dataset.
29
  It achieves the following results on the evaluation set:
30
+ - Loss: 0.4071
31
+ - Accuracy: 0.8612
32
+ - Precision: 0.8523
33
+ - Recall: 0.8748
34
+ - F1: 0.8630
35
 
36
  ## Model description
37
 
 
62
 
63
  | Training Loss | Epoch | Step | Validation Loss | Accuracy | Precision | Recall | F1 |
64
  |:-------------:|:-----:|:----:|:---------------:|:--------:|:---------:|:------:|:------:|
65
+ | 0.4097 | 1.0 | 714 | 0.4866 | 0.8362 | 0.8105 | 0.8674 | 0.8335 |
66
+ | 0.2884 | 2.0 | 1428 | 0.4015 | 0.8564 | 0.8476 | 0.8744 | 0.8593 |
67
+ | 0.2299 | 3.0 | 2142 | 0.4071 | 0.8612 | 0.8523 | 0.8748 | 0.8630 |
68
 
69
 
70
  ### Framework versions
71
 
72
+ - Transformers 4.54.0
73
  - Pytorch 2.6.0+cu124
74
  - Datasets 4.0.0
75
+ - Tokenizers 0.21.2
config.json CHANGED
@@ -41,7 +41,7 @@
41
  "summary_type": "first",
42
  "summary_use_proj": true,
43
  "torch_dtype": "float32",
44
- "transformers_version": "4.54.1",
45
  "type_vocab_size": 2,
46
  "use_cache": true,
47
  "vocab_size": 35000
 
41
  "summary_type": "first",
42
  "summary_use_proj": true,
43
  "torch_dtype": "float32",
44
+ "transformers_version": "4.54.0",
45
  "type_vocab_size": 2,
46
  "use_cache": true,
47
  "vocab_size": 35000
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:81534350b8195762c08fb72581ecb57ae1972e635dceed28bada6060f37e5708
3
  size 451731052
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7bd3f58da41994d1a1fbe7a1ab5b8173b0506818a0812f53b849ef4512d4c627
3
  size 451731052
training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:b2f4e1533860d18d9040930b270926ebfce8bc0787e9b3d5f5c8c11694ac3184
3
  size 5304
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:216c784e411c7b79843aa73f2484b3145ea142e607b867feaa440fc8ccbb4a9e
3
  size 5304