Update template_gradio_interface.py

template_gradio_interface.py

@@ -52,7 +52,16 @@ def call_predict(inference_dict, cols_order):
         y_pred_rescaled = scaler_targets.inverse_transform(y_pred)
 
         plt.clf()
-        shap.summary_plot(shap_values[0], feature_names=df_preprocessed.columns)
+        # shap.summary_plot(shap_values[0], feature_names=df_preprocessed.columns)
+        plt.figure(figsize=(15, 15))
+        plt.subplot(1,2,1)
+        shap.summary_plot(shap_values[0], input, show=False, feature_names=df_preprocessed.columns, plot_size=(15, 15))
+        plt.subplot(1,2,2)
+        shap.summary_plot(shap_values[1], input, show=False, feature_names=df_preprocessed.columns, plot_size=(15, 15))
+        plt.subplot(1,2,3)
+        shap.summary_plot(shap_values[2], input, show=False, feature_names=df_preprocessed.columns, plot_size=(15, 15))
+        plt.tight_layout()
+        plt.subplots_adjust(wspace=2.0)
         fig = plt.gcf()
 
         print("mmmmmmmmmmmmmmmmmmmmm")