synapti commited on
Commit
95130a1
·
verified ·
1 Parent(s): 2cbba05

Model save

Browse files
Files changed (2) hide show
  1. README.md +78 -0
  2. model.safetensors +1 -1
README.md ADDED
@@ -0,0 +1,78 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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-v2
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-v2
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.0026
25
+ - Accuracy: 0.9936
26
+ - F1: 0.9944
27
+ - Precision: 0.9889
28
+ - Recall: 1.0
29
+ - Roc Auc: 0.9989
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.0232 | 0.1305 | 100 | 0.0114 | 0.9496 | 0.9575 | 0.9272 | 0.9899 | 0.9948 |
65
+ | 0.0144 | 0.2609 | 200 | 0.0025 | 0.9925 | 0.9935 | 0.9890 | 0.9980 | 0.9997 |
66
+ | 0.0074 | 0.3914 | 300 | 0.0037 | 0.9948 | 0.9955 | 0.9960 | 0.9949 | 0.9996 |
67
+ | 0.0028 | 0.5219 | 400 | 0.0022 | 0.9971 | 0.9975 | 0.9960 | 0.9990 | 0.9995 |
68
+ | 0.002 | 0.6523 | 500 | 0.0038 | 0.9942 | 0.9950 | 0.9910 | 0.9990 | 0.9983 |
69
+ | 0.0004 | 0.7828 | 600 | 0.0023 | 0.9971 | 0.9975 | 0.9970 | 0.9980 | 0.9987 |
70
+ | 0.0052 | 0.9132 | 700 | 0.0008 | 0.9959 | 0.9965 | 0.9930 | 1.0 | 1.0000 |
71
+
72
+
73
+ ### Framework versions
74
+
75
+ - Transformers 4.57.3
76
+ - Pytorch 2.9.1+cu128
77
+ - Datasets 4.4.1
78
+ - Tokenizers 0.22.1
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:847c7811e4aa07980ce9ccebd19e7b168ae79e9f93a353777c5bc962fac9f411
3
  size 598439784
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:765834e01d81bc9fcc3440a9f65a0c9f826e718accb79335109d6940d16229af
3
  size 598439784