Update app.py

app.py

@@ -69,31 +69,24 @@ input_val = buil_UI(X)
 
 if st.button('Calculate House Price'):
     predictLGBM = model.predict(input_val)
+    # submissionLGBM = pd.DataFrame({'SalePrice':predictLGBM})
+    # result = submissionLGBM.iloc[0].values
     st.write(st.write('Predicted house price is: $',predictLGBM[0]))
-
+    
+    #input_val
     shap_values = shap.TreeExplainer(model).shap_values(input_val)
-    # st_shap(shap.plots.beeswarm(shap_values), height=300)
     fig_summary = shap.summary_plot(shap_values, input_val)
     st.pyplot(fig_summary)
 
     shap_interaction_values = shap.TreeExplainer(model).shap_interaction_values(input_val.iloc[:,:])
-    # st_shap(shap.summary_plot(shap_interaction_values, input_val.iloc[:,:]), height=400)
     interaction_fig = shap.summary_plot(shap_interaction_values, input_val.iloc[:,:])
     st.pyplot(interaction_fig)
-
-    # compute SHAP values
-    # explainer = shap.Explainer(model, input_val)
-    # shap_values = explainer(input_val)
-
-    # st_shap(shap.plots.waterfall(shap_values[0]), height=300)
-    # st_shap(shap.plots.beeswarm(shap_values), height=300)
-
-
-    # shap_interaction_values = shap.TreeExplainer(model).shap_interaction_values(input_val.iloc[:,:])
-    # shap.summary_plot(shap_interaction_values, input_val.iloc[:,:])
-
-
-
-# shap_values = shap.TreeExplainer(model,feature_perturbation="tree_path_dependent").shap_values(X)
-# st_shap(shap.force_plot(explainer.expected_value, shap_values[0,:], X_display.iloc[0,:]), height=200, width=1000)
-# st_shap(shap.force_plot(explainer.expected_value, shap_values[:1000,:], X_display.iloc[:1000,:]), height=400, width=1000)
+    
+    # X
+    shap_values_X = shap.TreeExplainer(model).shap_values(X)
+    fig_summary_X = shap.summary_plot(shap_values_X, X)
+    st.pyplot(fig_summary_X)
+
+    shap_interaction_values_X = shap.TreeExplainer(model).shap_interaction_values(X.iloc[:,:])
+    interaction_fig_X = shap.summary_plot(shap_interaction_values_X, X.iloc[:,:])
+    st.pyplot(interaction_fig_X)