ZivKassnerNK commited on
Commit
978938a
·
1 Parent(s): 7651061

Add evaluation results and output files to model card

Browse files
README.md CHANGED
@@ -6,8 +6,8 @@ tags:
6
  - evaluation
7
  metrics:
8
  - average_precision: 0.87
9
- - roc_auc: 0.76
10
- - best threshold according to F1: 0.13
11
  ---
12
 
13
  # Binary Classification Model
@@ -15,8 +15,8 @@ metrics:
15
  ## Evaluation Results
16
 
17
  **Average Precision:** 0.87
18
- **ROC AUC:** 0.76
19
- **best threshold according to F1: 0.13
20
 
21
 
22
  ## Visualizations
 
6
  - evaluation
7
  metrics:
8
  - average_precision: 0.87
9
+ - roc_auc: 0.75
10
+ - best threshold according to F1: 0.10
11
  ---
12
 
13
  # Binary Classification Model
 
15
  ## Evaluation Results
16
 
17
  **Average Precision:** 0.87
18
+ **ROC AUC:** 0.75
19
+ **best threshold according to F1: 0.10
20
 
21
 
22
  ## Visualizations
checkpoint-5/model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:48002186acebeeeb33e3af1a8bebc5b1d4d0df3112d77550512519a7308c4818
3
  size 8563512
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:70c5f1f80825f9e977f16d8b98dc58d27249ea261b2db50c1e233352e47f92ab
3
  size 8563512
checkpoint-5/optimizer.pt CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:5144dc3dd856c4e7e36106223acaa0e865befdb2ea2634f21bac36c0107d5272
3
  size 17171514
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ef5598f3384d225e894a29a0b6570189dd18658f674453591c2ac6e75e310bb9
3
  size 17171514
checkpoint-5/scheduler.pt CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:e5e2dd7b0b0aeae93e44a3701cca70fb38a7f111e82eb557413a5b28144c0332
3
  size 1000
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ce1f4ff52347a060a2285078af42de0c91fdecad7f6fa454864434eeabbb577b
3
  size 1000
checkpoint-5/trainer_state.json CHANGED
@@ -1,5 +1,5 @@
1
  {
2
- "best_metric": 0.8960872354073124,
3
  "best_model_checkpoint": "/tmp/logs/binary_classification_model_v3.1.5_junctions/checkpoint-5",
4
  "epoch": 1.0,
5
  "eval_steps": 500,
@@ -10,49 +10,49 @@
10
  "log_history": [
11
  {
12
  "epoch": 0.2,
13
- "grad_norm": 1199763.125,
14
- "learning_rate": 2.2614201718679333e-08,
15
- "loss": 0.4418,
16
  "step": 1
17
  },
18
  {
19
  "epoch": 0.4,
20
- "grad_norm": 1240740.375,
21
- "learning_rate": 4.5228403437358665e-08,
22
- "loss": 0.4503,
23
  "step": 2
24
  },
25
  {
26
  "epoch": 0.6,
27
- "grad_norm": 1050989.0,
28
- "learning_rate": 6.7842605156038e-08,
29
- "loss": 0.4108,
30
  "step": 3
31
  },
32
  {
33
  "epoch": 0.8,
34
- "grad_norm": 978190.3125,
35
- "learning_rate": 9.045680687471733e-08,
36
- "loss": 0.3993,
37
  "step": 4
38
  },
39
  {
40
  "epoch": 1.0,
41
- "grad_norm": 1178755.875,
42
- "learning_rate": 1.1307100859339666e-07,
43
- "loss": 0.4085,
44
  "step": 5
45
  },
46
  {
47
  "epoch": 1.0,
48
- "eval_accuracy": 0.8534599728629579,
49
- "eval_f1": 0.8960872354073124,
50
- "eval_loss": 0.42866843938827515,
51
- "eval_precision": 0.8769617074701821,
52
- "eval_recall": 0.9160655737704918,
53
- "eval_runtime": 2.9509,
54
- "eval_samples_per_second": 749.269,
55
- "eval_steps_per_second": 0.678,
56
  "step": 5
57
  }
58
  ],
 
1
  {
2
+ "best_metric": 0.8979460847240052,
3
  "best_model_checkpoint": "/tmp/logs/binary_classification_model_v3.1.5_junctions/checkpoint-5",
4
  "epoch": 1.0,
5
  "eval_steps": 500,
 
10
  "log_history": [
11
  {
12
  "epoch": 0.2,
13
+ "grad_norm": 1230799.5,
14
+ "learning_rate": 1.1307100859339666e-08,
15
+ "loss": 0.4355,
16
  "step": 1
17
  },
18
  {
19
  "epoch": 0.4,
20
+ "grad_norm": 1211314.625,
21
+ "learning_rate": 2.2614201718679333e-08,
22
+ "loss": 0.4418,
23
  "step": 2
24
  },
25
  {
26
  "epoch": 0.6,
27
+ "grad_norm": 1050969.25,
28
+ "learning_rate": 3.3921302578019e-08,
29
+ "loss": 0.3984,
30
  "step": 3
31
  },
32
  {
33
  "epoch": 0.8,
34
+ "grad_norm": 1020256.0,
35
+ "learning_rate": 4.5228403437358665e-08,
36
+ "loss": 0.3992,
37
  "step": 4
38
  },
39
  {
40
  "epoch": 1.0,
41
+ "grad_norm": 1193337.125,
42
+ "learning_rate": 5.653550429669833e-08,
43
+ "loss": 0.4199,
44
  "step": 5
45
  },
46
  {
47
  "epoch": 1.0,
48
+ "eval_accuracy": 0.8561736770691994,
49
+ "eval_f1": 0.8979460847240052,
50
+ "eval_loss": 0.4362623989582062,
51
+ "eval_precision": 0.8793211816467631,
52
+ "eval_recall": 0.9173770491803279,
53
+ "eval_runtime": 3.0211,
54
+ "eval_samples_per_second": 731.842,
55
+ "eval_steps_per_second": 0.662,
56
  "step": 5
57
  }
58
  ],
checkpoint-5/training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:93b8983655cf890e6632b5e9d2cba79a24784e85badd01191abefd8dc24ef793
3
  size 5368
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:848c632028e242f32d29c1d3dda7f29ecd3bd98ff1e36144536638cd67337a00
3
  size 5368
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:686976756f9cb128a41661bd80daea45d64527feb8db25f664f36f5f228ba273
3
  size 8563512
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:39fcdc5ff5a2d13e36b6937ba2e595c9d90bdcff5b58ade7a389cd7751016346
3
  size 8563512
pr_curve.png CHANGED
roc_curve.png CHANGED
training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:93b8983655cf890e6632b5e9d2cba79a24784e85badd01191abefd8dc24ef793
3
  size 5368
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:848c632028e242f32d29c1d3dda7f29ecd3bd98ff1e36144536638cd67337a00
3
  size 5368