Hugging Face's logo

lgk03
/
NDD-ppma_test-tags

Text Classification
Transformers
TensorBoard
Safetensors
distilbert
Generated from Trainer
Inference Endpoints
Model card Files Metrics Community
lgk03 commited on
Commit
5ea21c1
•
1 Parent(s): ce726f6

End of training

Browse files
Files changed (2) hide show
  1. README.md +7 -6
  2. model.safetensors +1 -1
README.md CHANGED
@@ -20,7 +20,7 @@ should probably proofread and complete it, then remove this comment. -->
20
 
21
  This model is a fine-tuned version of [distilbert-base-uncased](https://huggingface.co/distilbert-base-uncased) on an unknown dataset.
22
  It achieves the following results on the evaluation set:
23
- - Loss: 0.3977
24
  - Accuracy: 0.7922
25
  - F1: 0.8291
26
  - Precision: 0.9283
@@ -51,18 +51,19 @@ The following hyperparameters were used during training:
51
  - total_train_batch_size: 128
52
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
53
  - lr_scheduler_type: linear
54
- - num_epochs: 1
55
 
56
  ### Training results
57
 
58
  | Training Loss | Epoch | Step | Validation Loss | Accuracy | F1 | Precision | Recall |
59
  |:-------------:|:------:|:----:|:---------------:|:--------:|:------:|:---------:|:------:|
60
- | 0.1945 | 0.9990 | 722 | 0.3977 | 0.7922 | 0.8291 | 0.9283 | 0.7922 |
 
61
 
62
 
63
  ### Framework versions
64
 
65
- - Transformers 4.41.0
66
- - Pytorch 2.2.1+cu121
67
- - Datasets 2.19.1
68
  - Tokenizers 0.19.1
 
20
 
21
  This model is a fine-tuned version of [distilbert-base-uncased](https://huggingface.co/distilbert-base-uncased) on an unknown dataset.
22
  It achieves the following results on the evaluation set:
23
+ - Loss: 0.5434
24
  - Accuracy: 0.7922
25
  - F1: 0.8291
26
  - Precision: 0.9283
 
51
  - total_train_batch_size: 128
52
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
53
  - lr_scheduler_type: linear
54
+ - num_epochs: 2
55
 
56
  ### Training results
57
 
58
  | Training Loss | Epoch | Step | Validation Loss | Accuracy | F1 | Precision | Recall |
59
  |:-------------:|:------:|:----:|:---------------:|:--------:|:------:|:---------:|:------:|
60
+ | 0.1912 | 0.9990 | 722 | 0.3847 | 0.7922 | 0.8291 | 0.9283 | 0.7922 |
61
+ | 0.1512 | 1.9979 | 1444 | 0.5434 | 0.7922 | 0.8291 | 0.9283 | 0.7922 |
62
 
63
 
64
  ### Framework versions
65
 
66
+ - Transformers 4.41.2
67
+ - Pytorch 2.3.0+cu121
68
+ - Datasets 2.20.0
69
  - Tokenizers 0.19.1
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:4ebad3b9b51edcbf04c0e4c6bef40b112633ec1a99b422d98b6959a5999df315
3
  size 267832560
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d2dbd7bdc009ef7194cf813f3f43cfbbd213590da74443077de5cc00767c331e
3
  size 267832560