jonathanagustin commited on
Commit
4fd946e
1 Parent(s): e3b4663

Upload folder using huggingface_hub

Browse files
model.onnx CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:83eb6a35a587962db7174849f1bd9b44096bd1d063f2c5c4fa8869ed8944e269
3
  size 29639
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ff37406dd9bc008723236034757f16490c41c26d359164614dbccb9a40f43b78
3
  size 29639
neural_network/metrics/metrics.json CHANGED
@@ -1,25 +1,25 @@
1
  {
2
  "train": {
3
- "accuracy": 0.7377527952194214,
4
- "auc": 0.8147459626197815,
5
- "loss": 0.5141830444335938
6
  },
7
  "eval": {
8
- "accuracy": 0.7731721997261047,
9
- "auc": 0.6272097826004028,
10
- "loss": 0.5016754269599915
11
  },
12
  "step": 0,
13
- "val_accuracy": 0.7731721933679585,
14
- "val_precision": 0.19533136570968607,
15
- "val_recall": 0.32056362835755176,
16
- "val_f1": 0.24274758252750916,
17
- "val_auc": 0.6272456279863441,
18
- "val_avg_precision": 0.17174880265376707,
19
- "test_accuracy": 0.7703870162297128,
20
- "test_precision": 0.18009348364036293,
21
- "test_recall": 0.2884191985909291,
22
- "test_f1": 0.22173324306025727,
23
- "test_auc": 0.6234149477766672,
24
- "test_avg_precision": 0.1627787212516569
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
  }
neural_network/plots/confusion_matrix.png CHANGED
neural_network/plots/roc_curve.png CHANGED