2024/03/09/16:00

analysis/others/shap_model.py

@@ -7,7 +7,7 @@ from classes.static_custom_class import StaticValue
 
 def draw_shap_beeswarm(model, x, feature_names, type, paint_object):
     plt.clf()
-    x = shap.sample(x, min(66, len(x)), random_state=StaticValue.RANDOM_STATE)
+    x = shap.sample(x, min(20, len(x)), random_state=StaticValue.RANDOM_STATE)
     explainer = shap.KernelExplainer(model.predict, x)
     shap_values = explainer(x)
 
@@ -21,7 +21,7 @@ def draw_shap_beeswarm(model, x, feature_names, type, paint_object):
 
 def draw_waterfall(model, x, feature_names, number, paint_object):
     plt.clf()
-    x = shap.sample(x, min(66, len(x)), random_state=StaticValue.RANDOM_STATE)
+    x = shap.sample(x, min(20, len(x)), random_state=StaticValue.RANDOM_STATE)
     explainer = shap.KernelExplainer(model.predict, x, feature_names=feature_names)
     shap_values = explainer(x)
 
@@ -35,7 +35,7 @@ def draw_waterfall(model, x, feature_names, number, paint_object):
 
 def draw_force(model, x, feature_names, number, paint_object):
     plt.clf()
-    x = shap.sample(x, min(66, len(x)), random_state=StaticValue.RANDOM_STATE)
+    x = shap.sample(x, min(20, len(x)), random_state=StaticValue.RANDOM_STATE)
     explainer = shap.KernelExplainer(model.predict, x, feature_names=feature_names)
     shap_values = explainer(x[number])
 
@@ -49,7 +49,7 @@ def draw_force(model, x, feature_names, number, paint_object):
 
 def draw_dependence(model, x, feature_names, col, paint_object):
     plt.clf()
-    x = shap.sample(x, min(66, len(x)), random_state=StaticValue.RANDOM_STATE)
+    x = shap.sample(x, min(20, len(x)), random_state=StaticValue.RANDOM_STATE)
     explainer = shap.KernelExplainer(model.predict, x, feature_names=feature_names)
     shap_values = explainer(x)
 

app.py

@@ -1531,12 +1531,12 @@ def get_return(is_visible, extra_gr_dict: dict = None):
             data_fit_button: gr.Button(LN.data_fit_button, visible=Dataset.check_before_train()),
             waterfall_radio: gr.Radio(Dataset.get_trained_model_list(), visible=Dataset.check_before_train(),
                                       label=LN.waterfall_radio),
-            waterfall_number: gr.Slider(0, 66, value=0, step=1,
+            waterfall_number: gr.Slider(0, 20, value=0, step=1,
                                         visible=Dataset.check_before_train(), label=LN.waterfall_number),
             waterfall_button: gr.Button(LN.waterfall_button, visible=Dataset.check_before_train()),
             force_radio: gr.Radio(Dataset.get_trained_model_list(), visible=Dataset.check_before_train(),
                                   label=LN.force_radio),
-            force_number: gr.Slider(0, 66, value=0, step=1,
+            force_number: gr.Slider(0, 20, value=0, step=1,
                                     visible=Dataset.check_before_train(), label=LN.force_number),
             force_button: gr.Button(LN.force_button, visible=Dataset.check_before_train()),
             dependence_radio: gr.Radio(Dataset.get_trained_model_list(), visible=Dataset.check_before_train(),