ashield-ai commited on
Commit
2e5467e
·
verified ·
1 Parent(s): 4c6a4dd

End of training

Browse files
README.md ADDED
@@ -0,0 +1,65 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ library_name: transformers
3
+ license: apache-2.0
4
+ base_model: answerdotai/ModernBERT-base
5
+ tags:
6
+ - generated_from_trainer
7
+ metrics:
8
+ - f1
9
+ model-index:
10
+ - name: domain-classifier
11
+ results: []
12
+ ---
13
+
14
+ <!-- This model card has been generated automatically according to the information the Trainer had access to. You
15
+ should probably proofread and complete it, then remove this comment. -->
16
+
17
+ # domain-classifier
18
+
19
+ This model is a fine-tuned version of [answerdotai/ModernBERT-base](https://huggingface.co/answerdotai/ModernBERT-base) on an unknown dataset.
20
+ It achieves the following results on the evaluation set:
21
+ - Loss: 0.7974
22
+ - F1: 0.8946
23
+
24
+ ## Model description
25
+
26
+ More information needed
27
+
28
+ ## Intended uses & limitations
29
+
30
+ More information needed
31
+
32
+ ## Training and evaluation data
33
+
34
+ More information needed
35
+
36
+ ## Training procedure
37
+
38
+ ### Training hyperparameters
39
+
40
+ The following hyperparameters were used during training:
41
+ - learning_rate: 5e-05
42
+ - train_batch_size: 8
43
+ - eval_batch_size: 4
44
+ - seed: 42
45
+ - optimizer: Use adamw_torch_fused with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
46
+ - lr_scheduler_type: linear
47
+ - num_epochs: 5
48
+
49
+ ### Training results
50
+
51
+ | Training Loss | Epoch | Step | Validation Loss | F1 |
52
+ |:-------------:|:-----:|:----:|:---------------:|:------:|
53
+ | 0.0581 | 1.0 | 113 | 1.1200 | 0.8636 |
54
+ | 0.1466 | 2.0 | 226 | 0.9247 | 0.8918 |
55
+ | 0.0279 | 3.0 | 339 | 0.8111 | 0.8946 |
56
+ | 0.0131 | 4.0 | 452 | 0.7964 | 0.8946 |
57
+ | 0.0046 | 5.0 | 565 | 0.7974 | 0.8946 |
58
+
59
+
60
+ ### Framework versions
61
+
62
+ - Transformers 4.48.0
63
+ - Pytorch 2.5.1+cu121
64
+ - Datasets 3.2.0
65
+ - Tokenizers 0.21.0
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:ed37af2fdabc03bb82c54c2ac80592037e79e3bfb90050ebc25091cfaad512ad
3
  size 598513616
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f39511143b5ea546ce8a3322b2395fbb80526cfca3a72f1b144559dab1b59b42
3
  size 598513616
runs/Jan14_06-19-36_9ae06be23b84/events.out.tfevents.1736835577.9ae06be23b84.1197.2 CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:bf5b1be5841f961a1ca0bf394a14917c0ed25668070d3028af379d66335a3eb1
3
- size 9925
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a2a286470d6cfbf6d396ea60513e60cdf78ccf4d26f0d52535cbea98bc5aa284
3
+ size 10279