Hugging Face's logo

logiczmaksimka
/
finetuned_distilbert-base-uncased

Text Classification
Transformers
PyTorch
distilbert
Generated from Trainer
Inference Endpoints
Model card Files Community
logiczmaksimka commited on
Commit
65f03b4
•
1 Parent(s): c252d5c

End of training

Browse files
Files changed (4) hide show
  1. README.md +4 -4
  2. config.json +13 -13
  3. pytorch_model.bin +1 -1
  4. training_args.bin +1 -1
README.md CHANGED
@@ -17,8 +17,8 @@ should probably proofread and complete it, then remove this comment. -->
17
 
18
  This model is a fine-tuned version of [distilbert-base-uncased](https://huggingface.co/distilbert-base-uncased) on the None dataset.
19
  It achieves the following results on the evaluation set:
20
- - Loss: 0.7130
21
- - Accuracy: 0.7894
22
 
23
  ## Model description
24
 
@@ -49,8 +49,8 @@ The following hyperparameters were used during training:
49
 
50
  | Training Loss | Epoch | Step | Validation Loss | Accuracy |
51
  |:-------------:|:-----:|:----:|:---------------:|:--------:|
52
- | No log | 1.0 | 395 | 0.7772 | 0.7788 |
53
- | 1.202 | 2.0 | 790 | 0.6874 | 0.8040 |
54
 
55
 
56
  ### Framework versions
 
17
 
18
  This model is a fine-tuned version of [distilbert-base-uncased](https://huggingface.co/distilbert-base-uncased) on the None dataset.
19
  It achieves the following results on the evaluation set:
20
+ - Loss: 0.7476
21
+ - Accuracy: 0.7753
22
 
23
  ## Model description
24
 
 
49
 
50
  | Training Loss | Epoch | Step | Validation Loss | Accuracy |
51
  |:-------------:|:-----:|:----:|:---------------:|:--------:|
52
+ | No log | 1.0 | 395 | 0.8300 | 0.7663 |
53
+ | 1.2776 | 2.0 | 790 | 0.7021 | 0.7981 |
54
 
55
 
56
  ### Framework versions
config.json CHANGED
@@ -9,19 +9,19 @@
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
- "6": "LABEL_6",
19
- "7": "LABEL_7",
20
- "8": "LABEL_8",
21
- "9": "LABEL_9",
22
- "10": "LABEL_10",
23
- "11": "LABEL_11",
24
- "12": "LABEL_12"
25
  },
26
  "initializer_range": 0.02,
27
  "label2id": {
 
9
  "dropout": 0.1,
10
  "hidden_dim": 3072,
11
  "id2label": {
12
+ "0": "opinion",
13
+ "1": "appreciation",
14
+ "2": "complaint",
15
+ "3": "dev_command",
16
+ "4": "statement",
17
+ "5": "other_answers",
18
+ "6": "neg_answer",
19
+ "7": "comment",
20
+ "8": "open_question_factual",
21
+ "9": "open_question_opinion",
22
+ "10": "command",
23
+ "11": "yes_no_question",
24
+ "12": "pos_answer"
25
  },
26
  "initializer_range": 0.02,
27
  "label2id": {
pytorch_model.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:701fd675515d1c223a3eaa21c3686edec207495d35038dc9225c7c7f22468ef1
3
  size 267889770
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c00efc343f1fdf43f9a7f12051dc105a92804f76e028130f62e337fb7756d458
3
  size 267889770
training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:0ac6e19ccd6ecb81dd6cbcdd5ca4b876a09a5ce1a5ab422c0bef2276b18cea4d
3
  size 4536
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:73dde1fee856dbf79c67d579b118ebfda555fc7ea81181af6a4a1854e41f84e5
3
  size 4536