AurelPx commited on
Commit
667f60b
·
verified ·
1 Parent(s): 1c21d3c

Add metadata

Browse files
Files changed (1) hide show
  1. metadata.json +29 -0
metadata.json ADDED
@@ -0,0 +1,29 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "model_name": "IncomeSlayer-9000",
3
+ "task": "binary_classification",
4
+ "dataset": "Adult Income (OpenML task 7592)",
5
+ "n_samples": 48842,
6
+ "n_features_engineered": 28,
7
+ "cv_protocol": "10-fold stratified CV",
8
+ "results": {
9
+ "auc_roc": 0.93147,
10
+ "accuracy": 0.87599,
11
+ "auc_mean_fold": 0.9313,
12
+ "auc_std_fold": 0.00293
13
+ },
14
+ "openml_sota": {
15
+ "auc": 0.9284,
16
+ "accuracy": 0.874,
17
+ "method": "AdaBoost 2017"
18
+ },
19
+ "delta_vs_sota": {
20
+ "delta_auc": 0.00307,
21
+ "delta_accuracy": 0.00199
22
+ },
23
+ "ensemble_weights": {
24
+ "lgb": 0.1,
25
+ "xgb": 0.3,
26
+ "catboost": 0.6
27
+ },
28
+ "threshold": 0.512
29
+ }