synapti commited on
Commit
995b7a3
·
verified ·
1 Parent(s): 9365c56

Model save

Browse files
Files changed (2) hide show
  1. README.md +81 -0
  2. model.safetensors +1 -1
README.md ADDED
@@ -0,0 +1,81 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
+ - accuracy
9
+ - f1
10
+ - precision
11
+ - recall
12
+ model-index:
13
+ - name: nci-binary-detector
14
+ results: []
15
+ ---
16
+
17
+ <!-- This model card has been generated automatically according to the information the Trainer had access to. You
18
+ should probably proofread and complete it, then remove this comment. -->
19
+
20
+ # nci-binary-detector
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.0010
25
+ - Accuracy: 0.9977
26
+ - F1: 0.9980
27
+ - Precision: 0.9970
28
+ - Recall: 0.9990
29
+ - Roc Auc: 0.9999
30
+
31
+ ## Model description
32
+
33
+ More information needed
34
+
35
+ ## Intended uses & limitations
36
+
37
+ More information needed
38
+
39
+ ## Training and evaluation data
40
+
41
+ More information needed
42
+
43
+ ## Training procedure
44
+
45
+ ### Training hyperparameters
46
+
47
+ The following hyperparameters were used during training:
48
+ - learning_rate: 2e-05
49
+ - train_batch_size: 16
50
+ - eval_batch_size: 32
51
+ - seed: 42
52
+ - gradient_accumulation_steps: 2
53
+ - total_train_batch_size: 32
54
+ - optimizer: Use OptimizerNames.ADAMW_TORCH_FUSED with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
55
+ - lr_scheduler_type: linear
56
+ - lr_scheduler_warmup_ratio: 0.1
57
+ - num_epochs: 5
58
+ - mixed_precision_training: Native AMP
59
+
60
+ ### Training results
61
+
62
+ | Training Loss | Epoch | Step | Validation Loss | Accuracy | F1 | Precision | Recall | Roc Auc |
63
+ |:-------------:|:------:|:----:|:---------------:|:--------:|:------:|:---------:|:------:|:-------:|
64
+ | 0.011 | 0.1634 | 100 | 0.0047 | 0.9867 | 0.9883 | 0.9949 | 0.9818 | 0.9987 |
65
+ | 0.0017 | 0.3268 | 200 | 0.0046 | 0.9971 | 0.9975 | 0.9970 | 0.9980 | 0.9993 |
66
+ | 0.0004 | 0.4902 | 300 | 0.0028 | 0.9971 | 0.9975 | 0.9980 | 0.9970 | 0.9999 |
67
+ | 0.0106 | 0.6536 | 400 | 0.0008 | 0.9983 | 0.9985 | 0.9980 | 0.9990 | 1.0000 |
68
+ | 0.0001 | 0.8170 | 500 | 0.0011 | 0.9983 | 0.9985 | 0.9980 | 0.9990 | 1.0000 |
69
+ | 0.0007 | 0.9804 | 600 | 0.0012 | 0.9983 | 0.9985 | 0.9980 | 0.9990 | 1.0000 |
70
+ | 0.0 | 1.1438 | 700 | 0.0010 | 0.9988 | 0.9990 | 0.9980 | 1.0 | 1.0000 |
71
+ | 0.0021 | 1.3072 | 800 | 0.0006 | 0.9977 | 0.9980 | 0.9980 | 0.9980 | 1.0000 |
72
+ | 0.0018 | 1.4706 | 900 | 0.0010 | 0.9988 | 0.9990 | 0.9980 | 1.0 | 1.0000 |
73
+ | 0.0012 | 1.6340 | 1000 | 0.0017 | 0.9977 | 0.9980 | 0.9960 | 1.0 | 1.0000 |
74
+
75
+
76
+ ### Framework versions
77
+
78
+ - Transformers 4.57.3
79
+ - Pytorch 2.9.1+cu128
80
+ - Datasets 4.4.1
81
+ - Tokenizers 0.22.1
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:30014694760d63a9f83f0012563f3a2d507c3eb187c55f3122197a5e710255be
3
  size 598439784
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:fd84c754687a4ebed30993d95bdc2cdd0ce65ba26e9b50c8e0ba3afd6c974e56
3
  size 598439784