Par123321/Heart_Failure_Death_Detection

1import numpy as np2import pandas as pd3import streamlit as st4import joblib5import feature_engine6# import model7model = joblib.load(open("best_boosting_model.pkl", "rb"))8 9st.write('Insert feature to predict wheter or not a patient would die')10 11anaemia = st.selectbox(label='Anaemia', options=['Yes', 'No'])12creatinine_phosphokinase = st.number_input(label='Creatinine phosphokinase', min_value=0, max_value=10000, value=245, step=1)13diabetes = st.selectbox(label='Diabetes', options=['Yes', 'No'])14ejection_fraction = st.number_input(label='ejection fraction', min_value=0, max_value=100, value=30, step=1)15high_blood_pressure = st.selectbox(label='high blood pressure', options=['Yes', 'No'])16platelets = st.number_input(label='platelets', min_value=0, max_value=850000, value=2000, step=1)17serum_creatinine = st.number_input(label='serum creatinine', min_value=0, max_value=10, value=3, step=1)18serum_sodium = st.number_input(label='serum sodium', min_value=100, max_value=150, value=120, step=1)19sex = st.selectbox(label='Sex', options=['Female', 'Male'])20smoking = st.selectbox(label='smoking', options=['Yes', 'No'])21time = st.number_input(label='time', min_value=70, max_value=300, value=120, step=1)22 23 24data = pd.DataFrame({'anaemia': [anaemia],25                'creatinine_phosphokinase': [creatinine_phosphokinase],26                'diabetes': [diabetes],27                'ejection_fraction':[ejection_fraction],28                'high_blood_pressure': [high_blood_pressure],29                'platelets': [platelets],30                'serum_creatinine': [serum_creatinine],31                'serum_sodium': [serum_sodium],32                'sex': [sex],33                'smoking':[smoking],34                'time':[time]35})36data.replace({'Yes':1,'No':0},inplace=True)37data.replace({"Female":0,"Male":1},inplace=True)38 39 40clas = model.predict(data).tolist()[0]41 42 43 44# interpretation45st.write('Classification Result: ')46if clas == 0:47    st.text('Not Dead')48else:49    st.text('Dead')