Spaces:
Sleeping
Sleeping
File size: 2,651 Bytes
191f139 5fbb6c5 191f139 5fbb6c5 191f139 5fbb6c5 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 | {
"dataset": "aiml2021/obesity",
"feature_columns": [
"Age",
"Height",
"Weight",
"FCVC",
"NCP",
"CH2O",
"FAF",
"TUE",
"Gender_Female",
"Gender_Male",
"family_history_with_overweight_no",
"family_history_with_overweight_yes",
"FAVC_no",
"FAVC_yes",
"CAEC_Always",
"CAEC_Frequently",
"CAEC_Sometimes",
"CAEC_no",
"SMOKE_no",
"SMOKE_yes",
"SCC_no",
"SCC_yes",
"CALC_Always",
"CALC_Frequently",
"CALC_Sometimes",
"CALC_no",
"MTRANS_Automobile",
"MTRANS_Bike",
"MTRANS_Motorbike",
"MTRANS_Public_Transportation",
"MTRANS_Walking"
],
"classes": [
"Insufficient_Weight",
"Normal_Weight",
"Overweight_Level_I",
"Overweight_Level_II",
"Obesity_Type_I",
"Obesity_Type_II",
"Obesity_Type_III"
],
"n_train": 1688,
"n_test": 423,
"regressor": {
"name": "XGBRegressor",
"target": "BMI",
"metrics": {
"mae": 0.3749128417977634,
"r2": 0.987534374087826
},
"baseline_metrics": {
"ridge_mae": 0.5687129506345994,
"ridge_r2": 0.9910858695403162
}
},
"classifier": {
"name": "XGBClassifier",
"target": "NObeyesdad",
"metrics": {
"accuracy": 0.9574468085106383,
"macro_f1": 0.9562422503730887
},
"baseline_metrics": {
"logit_accuracy": 0.8628841607565012,
"logit_macro_f1": 0.8583933672964451
},
"per_class": {
"Insufficient_Weight": {
"precision": 0.9607843137254902,
"recall": 0.9074074074074074,
"f1": 0.9333333333333333,
"support": 54
},
"Normal_Weight": {
"precision": 0.84375,
"recall": 0.9310344827586207,
"f1": 0.8852459016393442,
"support": 58
},
"Overweight_Level_I": {
"precision": 0.971830985915493,
"recall": 0.9857142857142858,
"f1": 0.9787234042553191,
"support": 70
},
"Overweight_Level_II": {
"precision": 0.9833333333333333,
"recall": 0.9833333333333333,
"f1": 0.9833333333333333,
"support": 60
},
"Obesity_Type_I": {
"precision": 1.0,
"recall": 0.9846153846153847,
"f1": 0.9922480620155039,
"support": 65
},
"Obesity_Type_II": {
"precision": 0.9636363636363636,
"recall": 0.9137931034482759,
"f1": 0.9380530973451328,
"support": 58
},
"Obesity_Type_III": {
"precision": 0.9827586206896551,
"recall": 0.9827586206896551,
"f1": 0.9827586206896551,
"support": 58
}
}
}
} |