jonathanagustin commited on
Commit
b246112
1 Parent(s): 21fb681

Upload folder using huggingface_hub

Browse files
autogluon/model/leaderboard.csv CHANGED
@@ -1,3 +1,3 @@
1
  model,score_test,score_val,eval_metric,pred_time_test,pred_time_val,fit_time,pred_time_test_marginal,pred_time_val_marginal,fit_time_marginal,stack_level,can_infer,fit_order
2
- LightGBMBasic,0.6467403398083669,0.6342024393654344,roc_auc,0.03393197059631348,0.03538703918457031,1.146812915802002,0.03393197059631348,0.03538703918457031,1.146812915802002,1,True,1
3
- WeightedEnsemble_L2,0.6467403398083669,0.6342024393654344,roc_auc,0.0435791015625,0.035985708236694336,1.149925947189331,0.009647130966186523,0.0005986690521240234,0.0031130313873291016,2,True,2
 
1
  model,score_test,score_val,eval_metric,pred_time_test,pred_time_val,fit_time,pred_time_test_marginal,pred_time_val_marginal,fit_time_marginal,stack_level,can_infer,fit_order
2
+ LightGBMBasic,0.6467403398083669,0.6342024393654344,roc_auc,0.018140077590942383,0.02796173095703125,1.547741174697876,0.018140077590942383,0.02796173095703125,1.547741174697876,1,True,1
3
+ WeightedEnsemble_L2,0.6467403398083669,0.6342024393654344,roc_auc,0.0198361873626709,0.028901338577270508,1.551680564880371,0.0016961097717285156,0.0009396076202392578,0.003939390182495117,2,True,2
autogluon/model/learner.pkl CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:15ff306c94de1c67d7c49887f2ddda86584e9575df3afdac54b8c4008c615cfa
3
  size 6740
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:437d18898e0bd89c77ba61611c27dd8f8fee8398ed6f602760c6317943ff7d6e
3
  size 6740
autogluon/model/metadata.json CHANGED
@@ -277,7 +277,7 @@
277
  "semver": "3.0.2",
278
  "sentencepiece": "0.2.0",
279
  "seqeval": "1.2.2",
280
- "setuptools": "75.2.0",
281
  "shellingham": "1.5.4",
282
  "shortuuid": "1.0.13",
283
  "shtab": "1.7.1",
 
277
  "semver": "3.0.2",
278
  "sentencepiece": "0.2.0",
279
  "seqeval": "1.2.2",
280
+ "setuptools": "75.3.0",
281
  "shellingham": "1.5.4",
282
  "shortuuid": "1.0.13",
283
  "shtab": "1.7.1",
neural_network/metrics/metrics.json CHANGED
@@ -1,25 +1,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
  }
 
1
  {
2
  "train": {
3
+ "accuracy": 0.7406350374221802,
4
+ "auc": 0.8165216445922852,
5
+ "loss": 0.5119302868843079
6
  },
7
  "eval": {
8
+ "accuracy": 0.7928485870361328,
9
+ "auc": 0.6286745071411133,
10
+ "loss": 0.4915105700492859
11
  },
12
  "step": 0,
13
+ "val_accuracy": 0.7928485817019576,
14
+ "val_precision": 0.1992950977250881,
15
+ "val_recall": 0.27388815499779834,
16
+ "val_f1": 0.23071216617210683,
17
+ "val_auc": 0.6287202765737177,
18
+ "val_avg_precision": 0.17043888595948037,
19
+ "test_accuracy": 0.789712858926342,
20
+ "test_precision": 0.1830065359477124,
21
+ "test_recall": 0.24658740642888596,
22
+ "test_f1": 0.21009191521290566,
23
+ "test_auc": 0.625395846072259,
24
+ "test_avg_precision": 0.1622468308743334
25
  }
neural_network/model/model.keras CHANGED
Binary files a/neural_network/model/model.keras and b/neural_network/model/model.keras differ
 
neural_network/plots/confusion_matrix.png CHANGED
neural_network/plots/roc_curve.png CHANGED