{ | |
"sklearn": { | |
"columns": [ | |
"AgeCategory", | |
"Stroke", | |
"Diabetic_Yes", | |
"KidneyDisease", | |
"Smoking", | |
"SkinCancer", | |
"Is_Male", | |
"BMI", | |
"Asthma", | |
"Race_White", | |
"AlcoholDrinking", | |
"GenHealth" | |
], | |
"environment": [ | |
"scikit-learn=1.2.0" | |
], | |
"example_input":{ | |
"AgeCategory": [ | |
1, | |
2 | |
], | |
"Stroke": [ | |
0, | |
1 | |
], | |
"Diabetic_Yes": [ | |
1, | |
0 | |
], | |
"KidneyDisease": [ | |
1, | |
0 | |
], | |
"Smoking": [ | |
1, | |
0 | |
], | |
"SkinCancer": [ | |
1, | |
0 | |
], | |
"Is_Male": [ | |
1, | |
0 | |
], | |
"BMI": [ | |
1, | |
2 | |
], | |
"Asthma": [ | |
1, | |
0 | |
], | |
"Race_White": [ | |
0, | |
1 | |
], | |
"AlcoholDrinking": [ | |
1, | |
0 | |
], | |
"GenHealth": [ | |
2, | |
3 | |
] | |
}, | |
"model": { | |
"file": "rf_model_to_predict_heartDisease.pkl" | |
}, | |
"task": "tabular-classification" | |
} | |
} |