HeartDiseasePredictor / config.json
abhatt13's picture
Update config.json
98ee736
raw
history blame
1.9 kB
{
"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"
}
}