-
Notifications
You must be signed in to change notification settings - Fork 0
/
hattack_predictor.py
125 lines (67 loc) · 3.14 KB
/
hattack_predictor.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
# -*- coding: utf-8 -*-
"""
Created on Mon Feb 27 16:17:01 2023
@author: soham
"""
import streamlit as st
import numpy as np
import pandas as pd
import joblib
import sklearn
def predict(data):
clf = joblib.load("clf.sav")
data=np.array(data).reshape(1,-1)
return clf.predict(data)
st.title("Mortality predictor")
st.header("Using heart attack data to predict the possibility of you dying in the said follow up period")
st.subheader("Kindly enter your details so that we can give you your predictions")
first_name,last_name = st.columns(2)
f_name=first_name.text_input("First Name")
l_name=last_name.text_input("Last Name")
button=st.button("Start")
if button:
st.write(f"Welcome {f_name}")
#####################################
# now we let the user pass in parameters
# Age, Anaemia, Creatnine_phosphokinase,diabetes,
# ejection_fraction,high_blood_pressure,platelets,
# Serum_creatinine, serum_sodium, sex, smoking
# time, Death_event (target)
#For anaemia,diabetes,high_blood_pressure,sex,smoking we need to use a dict
yes_no_bool = {"Yes":1,"No":0}
male_female_bool = {"Male":1,"Female":0}
st.write("-------------------------------------------")
age,anaemia,creatinine_phosphokinase = st.columns(3)
age = age.number_input("Enter your age",min_value=10.00,max_value=100.00) # Maybe add some limits here
anaemia = anaemia.radio("Do you have anaemia",["Yes","No"])
anaemia = yes_no_bool[anaemia]
creatinine_phosphokinase= creatinine_phosphokinase.number_input("CPK enzyme levels (mcg/L)",step=1)
st.write("---------------------------------------------")
diabetes,ejection_fraction,high_blood_pressure = st.columns(3)
diabetes = diabetes.radio("Do you have diabetes",["Yes","No"])
diabetes = yes_no_bool[diabetes]
ejection_fraction = ejection_fraction.number_input("Percentage of blood leaving",step=1,min_value=0,max_value=100)
high_blood_pressure = high_blood_pressure.radio("Do you have Hypertension",["Yes","No"])
high_blood_pressure = yes_no_bool[high_blood_pressure]
st.write("---------------------------------------------")
platelets,serum_creatinine,serum_sodium = st.columns(3)
platelets = platelets.number_input("Platelets in the blood")
serum_creatinine= serum_creatinine.number_input("Level of creatinine (mg/dL)")
serum_sodium=serum_sodium.number_input("Level of sodium in the blood (mEq/L)",step=1)
st.write("---------------------------------------------")
sex,smoking,time = st.columns(3)
sex = sex.radio("Sex",["Male","Female"])
sex = male_female_bool[sex]
smoking = smoking.radio("Do you smoke?",["Yes","No"])
smoking = yes_no_bool[smoking]
time=time.number_input("Follow up period",step=1)
st.write("---------------------------------------------")
data_instance=[age,anaemia,creatinine_phosphokinase,diabetes,
ejection_fraction,high_blood_pressure,platelets,
serum_creatinine,serum_sodium,sex,smoking,time]
st.write("#### Predict the chances of a death event")
prediction_converter = {0:"Less probability of death",1:"More probability of death"}
if st.button("Predict"):
prediction=predict(data_instance)
prediction=prediction_converter[prediction[0]]
st.write(prediction)