|
|
|
|
|
"""app.ipynb |
|
|
|
|
|
Automatically generated by Colaboratory. |
|
|
|
|
|
Original file is located at |
|
|
https://colab.research.google.com/drive/1NiGtTsWgQO-_sqya-2Qexup3M8BcmfYI |
|
|
""" |
|
|
|
|
|
import joblib |
|
|
import pandas as pd |
|
|
import streamlit as st |
|
|
|
|
|
AgeCategory_new = {'18-24':1, |
|
|
'25-29':2, |
|
|
'30-34':3, |
|
|
'35-39':4, |
|
|
'40-44':5, |
|
|
'45-49':6, |
|
|
'50-54':7, |
|
|
'55-59':8, |
|
|
'60-64':9, |
|
|
'65-69':10, |
|
|
'70-74':11, |
|
|
'75-79':12, |
|
|
'80 or older':13} |
|
|
|
|
|
model = joblib.load('model.joblib') |
|
|
unique_values = joblib.load('unique_values.joblib') |
|
|
|
|
|
unique_Smoking = unique_values["Smoking"] |
|
|
unique_AlcoholDrinking = unique_values["AlcoholDrinking"] |
|
|
unique_Stroke = unique_values["Stroke"] |
|
|
unique_DiffWalking = unique_values["DiffWalking"] |
|
|
unique_Sex = unique_values["Sex"] |
|
|
unique_race = unique_values["race"] |
|
|
unique_Diabetic = unique_values["Diabetic"] |
|
|
unique_PhysicalActivity = unique_values["PhysicalActivity"] |
|
|
unique_GenHealth = unique_values["GenHealth"] |
|
|
unique_Asthma = unique_values["Asthma"] |
|
|
unique_KidneyDisease = unique_values["KidneyDisease"] |
|
|
unique_SkinCancer = unique_values["SkinCancer"] |
|
|
unique_AgeCategory = unique_values["AgeCategory"] |
|
|
|
|
|
def main(): |
|
|
st.title("Personal Key Indicators of Heart Disease") |
|
|
|
|
|
with st.form("questionaire"): |
|
|
BMI = st.slider("BMI", min_value=10, max_value=100) |
|
|
Smoking = st.selectbox("Smoking", unique_Smoking) |
|
|
AlcoholDrinking = ("AlcoholDrinking", unique_AlcoholDrinking) |
|
|
Stroke = ("Stroke", unique_Stroke) |
|
|
PhysicalHealth = st.slider("PhysicalHealth", min_value=0, max_value=50) |
|
|
MentalHealth = st.slider("MentalHealth", min_value=0, max_value=50) |
|
|
DiffWalking = ("DiffWalking", unique_DiffWalking) |
|
|
Sex = ("Sex", unique_Sex) |
|
|
AgeCategory = ("AgeCategory_new", unique_AgeCategory) |
|
|
Race = ("Race", unique_Race) |
|
|
Diabetic = ("Diabetic", unique_Diabetic) |
|
|
PhysicalActivity = ("PhysicalActivity", unique_PhysicalActivity) |
|
|
GenHealth = ("GenHealth", unique_GenHealth) |
|
|
SleepTime = st.slider("SleepTime", min_value=0, max_value=24) |
|
|
Asthma = ("Asthma", unique_Asthma) |
|
|
KidneyDisease = ("KidneyDisease", unique_KidneyDisease) |
|
|
SkinCancer = ("SkinCancer", unique_SkinCancer) |
|
|
|
|
|
clicked = st.form_submit_button("Predict HeartDisease") |
|
|
if clicked: |
|
|
result=model.predict(pd.DataFrame({"BMI": [BMI], |
|
|
"Smoking": [Smoking], |
|
|
"AlcoholDrinking": [AlcoholDrinking], |
|
|
"Stroke": [Stroke], |
|
|
"PhysicalHealth": [PhysicalHealth], |
|
|
"MentalHealth": [MentalHealth], |
|
|
"DiffWalking": [DiffWalking], |
|
|
"sex": [sex], |
|
|
"AgeCategory_new": [AgeCategory_new[AgeCategory]], |
|
|
"race": [race], |
|
|
"Diabetic": [Diabetic], |
|
|
"PhysicalActivity": [PhysicalActivity], |
|
|
"GenHealth": [GenHealth], |
|
|
"SleepTime": [SleepTime], |
|
|
"Asthma": [Asthma], |
|
|
"KidneyDisease": [KidneyDisease], |
|
|
"SkinCancer": [SkinCancer]})) |
|
|
result = 'Yes' if result[0] == 1 else 'No' |
|
|
st.success('The predicted HeartDisease is {}'.format(result)) |
|
|
|
|
|
if __name__=='__main__': |
|
|
main() |
