Upload 3 files

app.py

@@ -0,0 +1,103 @@
+import pickle
+import pandas as pd
+import shap
+from shap.plots._force_matplotlib import draw_additive_plot
+import gradio as gr
+import numpy as np
+import matplotlib.pyplot as plt
+
+# load the model from disk
+loaded_model = pickle.load(open("heart_xgb.pkl", 'rb'))
+
+# Setup SHAP
+explainer = shap.Explainer(loaded_model) # PLEASE DO NOT CHANGE THIS.
+
+# Create the main function for server
+def main_func(age, sex, cp, trtbps, chol, fbs, restecg, thalachh,exng,oldpeak,slp,caa,thall):
+    new_row = pd.DataFrame.from_dict({'age':age,'sex':sex,
+              'cp':cp,'trtbps':trtbps,'chol':chol,
+              'fbs':fbs, 'restecg':restecg,'thalachh':thalachh,'exng':exng,
+                                     'oldpeak':oldpeak,'slp':slp,'caa':caa,'thall':thall}, 
+                                     orient = 'index').transpose()
+    
+    prob = loaded_model.predict_proba(new_row)
+    
+    shap_values = explainer(new_row)
+    # plot = shap.force_plot(shap_values[0], matplotlib=True, figsize=(30,30), show=False)
+    # plot = shap.plots.waterfall(shap_values[0], max_display=6, show=False)
+    plot = shap.plots.bar(shap_values[0], max_display=6, order=shap.Explanation.abs, show_data='auto', show=False)
+
+    plt.tight_layout()
+    local_plot = plt.gcf()
+    plt.close()
+    
+    return {"Low Chance": float(prob[0][0]), "High Chance": 1-float(prob[0][0])}, local_plot\
+
+# Create the UI
+title = "**Heart Attack Predictor & Interpreter** 🪐"
+description1 = """This app takes info from subjects and predicts their heart attack likelihood. Do not use for medical diagnosis."""
+
+description2 = """
+To use the app, click on one of the examples, or adjust the values of the factors, and click on Analyze. 🤞
+""" 
+
+with gr.Blocks(title=title) as demo:
+    gr.Markdown(f"## {title}")
+    gr.Markdown(description2)
+    gr.Markdown("""---""")
+    gr.Markdown(description2)
+    gr.Markdown("""---""")
+
+    with gr.Row():
+        age = gr.Number(label="age Score", value=40)
+
+    with gr.Row():        
+        with gr.Column():
+            sex = gr.Dropdown(label = "Sex", choices = ["Female", "Male"], type = "index" )
+            cp = gr.Dropdown(label="CP (Chest Pain) Score", choices = ["Typical Angina", "Atypical Angina", "Non-Anginal", "Asymptomatic"], type = "index", )
+            
+        with gr.Column():
+            cp = gr.Dropdown(label="cp Score", choices = ["Typical", "Atypical", "Non-Anginal", "Asymptomatic"], type = "index", 
+                    )
+            trtbps = gr.Number(label="trtbps Score", info = "resting blood pressure (in mm Hg)")
+
+    with gr.Row():   
+        with gr.Column():
+            fbs = gr.Slider(label="fbs Score", minimum=0, maximum=1, value=1, step=1)
+        with gr.Column():
+            restecg = gr.Slider(label="restecg Score", minimum=0, maximum=1, value=1, step=1)
+
+    with gr.Row():
+        with gr.Column():
+            thalachh = gr.Number(label="thalachh Score", info = "Maximum heart rate")
+        with gr.Column():
+            chol = gr.Number(label="chol Score", info = " cholestoral in mg/dl fetched via BMI sensor" )
+            exng = gr.Slider(label="exng Score", minimum=1, maximum=5, value=4, step=1)
+
+    
+    with gr.Row():      
+        with gr.Column():
+            oldpeak = gr.Slider(label="oldpeak Score", minimum=1, maximum=5, value=4, step=1)
+        with gr.Column():
+            slp = gr.Slider(label="slp Score", minimum=1, maximum=5, value=4, step=1)
+        with gr.Column():
+            caa = gr.Slider(label="caa Score", minimum=1, maximum=5, value=4, step=1)
+        with gr.Column():
+            thall = gr.Slider(label="thall Score", minimum=1, maximum=5, value=4, step=1)
+            
+            submit_btn = gr.Button("Analyze")
+
+    with gr.Column(visible=True) as output_col:
+            label = gr.Label(label = "Predicted Label")
+            local_plot = gr.Plot(label = 'Shap:')
+    
+    submit_btn.click(
+            main_func,
+            [age, sex, cp, trtbps, chol, fbs, restecg, thalachh,exng,oldpeak,slp,caa,thall],
+            [label,local_plot], api_name="Heart_Predictor"
+        )
+    
+    gr.Markdown("### Click on any of the examples below to see how it works:")
+        
+    
+demo.launch()

heart_xgb.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:f3d7f96dab791241a7536e789aadc37358ad1ccf276dccd89885ca8c2fdbdd4f
+size 132944

requirements.txt

@@ -0,0 +1,10 @@
+gradio==3.1.3
+Pillow
+yake
+pandas
+sklearn
+shap
+xgboost
+matplotlib
+numpy
+streamlit