sbaru/news-topic-classifier
Browse files- README.md +10 -10
- config.json +1 -1
- model.safetensors +1 -1
- 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.
|
| 31 |
-
- Accuracy: 0.
|
| 32 |
-
- Precision: 0.
|
| 33 |
-
- Recall: 0.
|
| 34 |
-
- F1: 0.
|
| 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.
|
| 66 |
-
| 0.2884 | 2.0 | 1428 | 0.
|
| 67 |
-
| 0.
|
| 68 |
|
| 69 |
|
| 70 |
### Framework versions
|
| 71 |
|
| 72 |
-
- Transformers 4.54.
|
| 73 |
- Pytorch 2.6.0+cu124
|
| 74 |
- Datasets 4.0.0
|
| 75 |
-
- Tokenizers 0.21.
|
|
|
|
| 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.
|
| 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:
|
| 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:
|
| 3 |
size 5304
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:216c784e411c7b79843aa73f2484b3145ea142e607b867feaa440fc8ccbb4a9e
|
| 3 |
size 5304
|