Spaces:
Sleeping
Sleeping
import joblib | |
import pandas as pd | |
import streamlit as st | |
ANAE_DIC = {'NO': 0, 'YES': 1} | |
DIA_DIC = {'NO': 0, 'YES': 1} | |
HIGHBLOOD_DIC = {'NO': 0, 'YES': 1} | |
SMOKE_DIC = {'NO': 0, 'YES': 1} | |
DEATH_DIC = {'NO': 0, 'YES': 1} | |
model = joblib.load('model (1).joblib') | |
unique_values = joblib.load('unique_values (1).joblib') | |
unique_age = unique_values["age"] | |
unique_anaemia = unique_values["anaemia"] | |
unique_creatinine = unique_values["creatinine_phosphokinase"] | |
unique_diabetes = unique_values["diabetes"] | |
unique_ejection = unique_values["ejection_fraction"] | |
unique_high_blood = unique_values["high_blood_pressure"] | |
unique_platelets = unique_values["platelets"] | |
unique_serum_creatinine = unique_values["serum_creatinine"] | |
unique_serum_sodium = unique_values["serum_sodium"] | |
unique_sex = unique_values["sex"] | |
unique_smoking = unique_values["smoking"] | |
unique_time = unique_values["time"] | |
def main(): | |
st.title("Adult Income Analysis") | |
with st.form("questionaire"): | |
age = st.slider("Age", min_value=10, max_value=100) | |
anaemia = st.selectbox("Anaemia", unique_anaemia) | |
creatinine = st.slider("Creatinine", min_value=10, max_value=10000) | |
diabetes = st.selectbox("Diabetes", unique_diabetes) | |
ejection = st.slider("Ejection(Percent)", min_value=1, max_value=100) | |
high_blood = st.selectbox("High blood pressure", unique_high_blood) | |
platelets = st.slider("Platelets", min_value=10000, max_value=1000000) | |
serum_creatinine = st.slider("Serum creatinine", min_value=0, max_value=10) | |
sodium = st.slider("Serum Sodium", min_value=100, max_value=200) | |
sex = st.selectbox("Sex", unique_sex) | |
smoking = st.selectbox("Smoke", unique_smoking) | |
time = st.slider("Time", min_value=10, max_value=100) | |
clicked = st.form_submit_button("Predict income") | |
if clicked: | |
result=model.predict(pd.DataFrame({"age": [age], | |
"anaemia": [ANAE_DIC[anaemia]], | |
"creatinine": [creatinine], | |
"diabetes": [DIA_DIC[diabetes]], | |
"ejection": [ejection], | |
"high_blood": [HIGHBLOOD_DIC[high_blood]], | |
"platelets": [platelets], | |
"serum_creatinine": [serum_creatinine], | |
"serum_sodium": [sodium], | |
"sex": [sex], | |
"smoking": [SMOKE_DIC[smoking]], | |
"time": [time]})) | |
result = 'No' if result[0] == 1 else 'Yes' | |
st.success('The predicted Death event is {}'.format(result)) | |
if __name__=='__main__': | |
main() | |