elihoole commited on
Commit
54757e9
·
verified ·
1 Parent(s): eb9052a

Add model trained for query classification (v4.0)

Browse files
README.md CHANGED
@@ -3,6 +3,7 @@ library_name: transformers
3
  license: apache-2.0
4
  base_model: answerdotai/ModernBERT-base
5
  tags:
 
6
  - generated_from_trainer
7
  metrics:
8
  - accuracy
@@ -21,11 +22,11 @@ should probably proofread and complete it, then remove this comment. -->
21
 
22
  This model is a fine-tuned version of [answerdotai/ModernBERT-base](https://huggingface.co/answerdotai/ModernBERT-base) on an unknown dataset.
23
  It achieves the following results on the evaluation set:
24
- - Loss: 0.1951
25
- - Accuracy: 0.9713
26
- - F1: 0.9713
27
- - Precision: 0.9724
28
- - Recall: 0.9713
29
 
30
  ## Model description
31
 
@@ -56,11 +57,11 @@ The following hyperparameters were used during training:
56
 
57
  | Training Loss | Epoch | Step | Validation Loss | Accuracy | F1 | Precision | Recall |
58
  |:-------------:|:-----:|:----:|:---------------:|:--------:|:------:|:---------:|:------:|
59
- | No log | 1.0 | 305 | 0.2426 | 0.9521 | 0.9520 | 0.9565 | 0.9521 |
60
- | 0.1447 | 2.0 | 610 | 0.1809 | 0.9598 | 0.9600 | 0.9618 | 0.9598 |
61
- | 0.1447 | 3.0 | 915 | 0.1568 | 0.9674 | 0.9676 | 0.9679 | 0.9674 |
62
- | 0.0158 | 4.0 | 1220 | 0.2314 | 0.9674 | 0.9675 | 0.9694 | 0.9674 |
63
- | 0.0022 | 5.0 | 1525 | 0.1951 | 0.9713 | 0.9713 | 0.9724 | 0.9713 |
64
 
65
 
66
  ### Framework versions
 
3
  license: apache-2.0
4
  base_model: answerdotai/ModernBERT-base
5
  tags:
6
+ - v4.0
7
  - generated_from_trainer
8
  metrics:
9
  - accuracy
 
22
 
23
  This model is a fine-tuned version of [answerdotai/ModernBERT-base](https://huggingface.co/answerdotai/ModernBERT-base) on an unknown dataset.
24
  It achieves the following results on the evaluation set:
25
+ - Loss: 0.1555
26
+ - Accuracy: 0.9789
27
+ - F1: 0.9790
28
+ - Precision: 0.9792
29
+ - Recall: 0.9789
30
 
31
  ## Model description
32
 
 
57
 
58
  | Training Loss | Epoch | Step | Validation Loss | Accuracy | F1 | Precision | Recall |
59
  |:-------------:|:-----:|:----:|:---------------:|:--------:|:------:|:---------:|:------:|
60
+ | No log | 1.0 | 305 | 0.2230 | 0.9579 | 0.9579 | 0.9600 | 0.9579 |
61
+ | 0.1385 | 2.0 | 610 | 0.1555 | 0.9789 | 0.9790 | 0.9792 | 0.9789 |
62
+ | 0.1385 | 3.0 | 915 | 0.1744 | 0.9693 | 0.9694 | 0.9701 | 0.9693 |
63
+ | 0.0189 | 4.0 | 1220 | 0.2378 | 0.9674 | 0.9675 | 0.9684 | 0.9674 |
64
+ | 0.0022 | 5.0 | 1525 | 0.2181 | 0.9732 | 0.9733 | 0.9737 | 0.9732 |
65
 
66
 
67
  ### Framework versions
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:1837ef2ef307669b9915a363d06d930d8c24af70dd8a16a8bd73cf9ac4b75743
3
  size 598445936
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:cd210befdccf108532d64e0919a3c22b08b20e1c4279cd18c7d677261a85152e
3
  size 598445936
runs/Apr10_11-49-18_ef403fb20d9e/events.out.tfevents.1744285766.ef403fb20d9e.188.2 CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:cc2597bacab43c07770a8134a05310cd166f0e75c3752159efd3117ab2c004d2
3
- size 9055
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8cddaf8825bba1832d4b8ccf95136ab1b8a68826a7765a2c1c5cd84bcc645b4a
3
+ size 9409
runs/Apr10_11-49-18_ef403fb20d9e/events.out.tfevents.1744286461.ef403fb20d9e.188.3 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:89f4eb10282a98c853763d6d6495a6ede04605d02efe97c01c397cd03fd3aaba
3
+ size 1032