HeartDiseasePredictor / config.json
abhatt13's picture
Update config.json
243a2bf
raw
history blame
1.95 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": [
"0",
"1"
],
"Stroke": [
"0",
"1"
],
"Diabetic_Yes": [
"1",
"0"
],
"KidneyDisease": [
"1",
"0"
],
"Smoking": [
"1",
"0"
],
"SkinCancer": [
"1",
"0"
],
"Is_Male": [
"1",
"0"
],
"BMI": [
"0",
"1"
],
"Asthma": [
"1",
"0"
],
"Race_White": [
"0",
"1"
],
"AlcoholDrinking": [
"1",
"0"
],
"GenHealth": [
"0",
"1"
]
},
"model": {
"file": "rf_model_to_predict_heartDisease.pkl"
},
"task": "tabular-classification"
}
}