jonathanagustin commited on
Commit
51bc8c1
1 Parent(s): 1857a5c

Upload folder using huggingface_hub

Browse files
neural_network/metrics/metrics.json CHANGED
@@ -1,25 +1,25 @@
1
  {
2
  "train": {
3
- "accuracy": 0.7387104034423828,
4
- "auc": 0.8177250027656555,
5
- "loss": 0.5098954439163208
6
  },
7
  "eval": {
8
- "accuracy": 0.7939472794532776,
9
- "auc": 0.629897952079773,
10
- "loss": 0.49378007650375366
11
  },
12
  "step": 0,
13
- "val_accuracy": 0.7939472632840592,
14
- "val_precision": 0.20933876527734252,
15
- "val_recall": 0.29414354909731394,
16
- "val_f1": 0.24459904796777737,
17
- "val_auc": 0.6298794259889418,
18
- "val_avg_precision": 0.17467399109502613,
19
- "test_accuracy": 0.7883645443196005,
20
- "test_precision": 0.18648390181702262,
21
- "test_recall": 0.2575957727873184,
22
- "test_f1": 0.21634615384615385,
23
- "test_auc": 0.6225499657310808,
24
- "test_avg_precision": 0.1635421080438205
25
  }
 
1
  {
2
  "train": {
3
+ "accuracy": 0.7367294430732727,
4
+ "auc": 0.8155916333198547,
5
+ "loss": 0.5119379758834839
6
  },
7
  "eval": {
8
+ "accuracy": 0.7842089533805847,
9
+ "auc": 0.6276553869247437,
10
+ "loss": 0.49857422709465027
11
  },
12
  "step": 0,
13
+ "val_accuracy": 0.7842089492608869,
14
+ "val_precision": 0.2004675628287551,
15
+ "val_recall": 0.3020695728753853,
16
+ "val_f1": 0.24099771649393992,
17
+ "val_auc": 0.6277547176489517,
18
+ "val_avg_precision": 0.1731221269461989,
19
+ "test_accuracy": 0.7796754057428215,
20
+ "test_precision": 0.18319029298609055,
21
+ "test_recall": 0.27256715103478646,
22
+ "test_f1": 0.2191150442477876,
23
+ "test_auc": 0.6229996507382787,
24
+ "test_avg_precision": 0.16467685639778956
25
  }
neural_network/model/model.onnx CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:ff37406dd9bc008723236034757f16490c41c26d359164614dbccb9a40f43b78
3
  size 29639
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a01116cd51ada33af6e180b2aab9c5053b03c7cc5b43320caba97736fcc323a8
3
  size 29639
neural_network/plots/confusion_matrix.png CHANGED
neural_network/plots/roc_curve.png CHANGED