| | |
| | """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 = st.selectbox("AlcoholDrinking", unique_AlcoholDrinking) |
| | Stroke = st.selectbox("Stroke", unique_Stroke) |
| | PhysicalHealth = st.slider("PhysicalHealth", min_value=0, max_value=50) |
| | MentalHealth = st.slider("MentalHealth", min_value=0, max_value=50) |
| | DiffWalking = st.selectbox("DiffWalking", unique_DiffWalking) |
| | Sex = st.selectbox("Sex", unique_Sex) |
| | AgeCategory = st.selectbox("AgeCategory_new", unique_AgeCategory) |
| | Race = st.selectbox("Race", unique_Race) |
| | Diabetic = st.selectbox("Diabetic", unique_Diabetic) |
| | PhysicalActivity = st.selectbox("PhysicalActivity", unique_PhysicalActivity) |
| | GenHealth = st.selectbox("GenHealth", unique_GenHealth) |
| | SleepTime = st.slider("SleepTime", min_value=0, max_value=24) |
| | Asthma = st.selectbox("Asthma", unique_Asthma) |
| | KidneyDisease = st.selectbox("KidneyDisease", unique_KidneyDisease) |
| | SkinCancer = st.selectbox("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": [AgeCategory_new[AgeCategory]], |
| | "Race": [Race], |
| | "Diabetic": [Diabetic], |
| | "PhysicalActivity": [PhysicalActivity], |
| | "GenHealth": [GenHealth], |
| | "SleepTime": [SleepTime], |
| | "Asthma": [Asthma], |
| | "KidneyDisease": [KidneyDisease], |
| | "SkinCancer": [SkinCancer]})) |
| | result = 'You are at risk of developing heart disease.' if result[0] == 1 else 'You are not at risk of heart disease.' |
| | st.success('The predicted HeartDisease is {}'.format(result)) |
| |
|
| | if __name__=='__main__': |
| | main() |
