Hugging Face's logo

natutaro
/
line-distilbert-base-japanese-finetuned-emotion

Text Classification
Transformers
TensorBoard
Safetensors
distilbert
Generated from Trainer
Inference Endpoints
Model card Files Metrics Community
natutaro commited on
Commit
b68e210
·
verified ·
1 Parent(s): a6ff9f6

Training completed!

Browse files
Files changed (5) hide show
  1. README.md +5 -5
  2. config.json +0 -16
  3. model.safetensors +2 -2
  4. runs/Jan25_23-51-40_06953ef1bb1d/events.out.tfevents.1706226770.06953ef1bb1d.781.0 +3 -0
  5. training_args.bin +1 -1
README.md CHANGED
@@ -18,9 +18,9 @@ should probably proofread and complete it, then remove this comment. -->
18
 
19
  This model is a fine-tuned version of [line-corporation/line-distilbert-base-japanese](https://huggingface.co/line-corporation/line-distilbert-base-japanese) on the None dataset.
20
  It achieves the following results on the evaluation set:
21
- - Loss: 0.2371
22
- - Accuracy: 0.911
23
- - F1: 0.9110
24
 
25
  ## Model description
26
 
@@ -51,8 +51,8 @@ The following hyperparameters were used during training:
51
 
52
  | Training Loss | Epoch | Step | Validation Loss | Accuracy | F1 |
53
  |:-------------:|:-----:|:----:|:---------------:|:--------:|:------:|
54
- | 0.502 | 1.0 | 125 | 0.2502 | 0.904 | 0.9040 |
55
- | 0.2184 | 2.0 | 250 | 0.2371 | 0.911 | 0.9110 |
56
 
57
 
58
  ### Framework versions
 
18
 
19
  This model is a fine-tuned version of [line-corporation/line-distilbert-base-japanese](https://huggingface.co/line-corporation/line-distilbert-base-japanese) on the None dataset.
20
  It achieves the following results on the evaluation set:
21
+ - Loss: 0.2180
22
+ - Accuracy: 0.92
23
+ - F1: 0.9200
24
 
25
  ## Model description
26
 
 
51
 
52
  | Training Loss | Epoch | Step | Validation Loss | Accuracy | F1 |
53
  |:-------------:|:-----:|:----:|:---------------:|:--------:|:------:|
54
+ | 0.3356 | 1.0 | 125 | 0.2241 | 0.918 | 0.918 |
55
+ | 0.1974 | 2.0 | 250 | 0.2180 | 0.92 | 0.9200 |
56
 
57
 
58
  ### Framework versions
config.json CHANGED
@@ -8,23 +8,7 @@
8
  "dim": 768,
9
  "dropout": 0.1,
10
  "hidden_dim": 3072,
11
- "id2label": {
12
- "0": "LABEL_0",
13
- "1": "LABEL_1",
14
- "2": "LABEL_2",
15
- "3": "LABEL_3",
16
- "4": "LABEL_4",
17
- "5": "LABEL_5"
18
- },
19
  "initializer_range": 0.02,
20
- "label2id": {
21
- "LABEL_0": 0,
22
- "LABEL_1": 1,
23
- "LABEL_2": 2,
24
- "LABEL_3": 3,
25
- "LABEL_4": 4,
26
- "LABEL_5": 5
27
- },
28
  "max_position_embeddings": 512,
29
  "model_type": "distilbert",
30
  "n_heads": 12,
 
8
  "dim": 768,
9
  "dropout": 0.1,
10
  "hidden_dim": 3072,
 
 
 
 
 
 
 
 
11
  "initializer_range": 0.02,
 
 
 
 
 
 
 
 
12
  "max_position_embeddings": 512,
13
  "model_type": "distilbert",
14
  "n_heads": 12,
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:8562ca38adfbbcfec74a9a31a4d9ee4a19a089538643f2c7843e293057dbbfe4
3
- size 274744592
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:994499add796222a000571230c5114536d3af18bf8a787076906920ce88e6a6c
3
+ size 274732280
runs/Jan25_23-51-40_06953ef1bb1d/events.out.tfevents.1706226770.06953ef1bb1d.781.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7cda76a36f30d5b66c79fd43d675dc25ee2a10d4fc0f96ed47d7dafb6af5da31
3
+ size 5683
training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:d86f424ab46112662120d902c82ff30b4c4a11286306142e8be37b553ff4f3dc
3
  size 4664
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e545ac931a02839d0577678ae2d744f131c057b6869500c9cbac4e36e8c1ba8c
3
  size 4664