2024/03/09/15: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(123, len(x)), random_state=StaticValue.RANDOM_STATE)
+    x = shap.sample(x, min(60, 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(123, len(x)), random_state=StaticValue.RANDOM_STATE)
+    x = shap.sample(x, min(60, len(x)), random_state=StaticValue.RANDOM_STATE)
     explainer = shap.KernelExplainer(model.predict, x, feature_names=feature_names)
     shap_values = explainer(x)
 
@@ -48,7 +48,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(123, len(x)), random_state=StaticValue.RANDOM_STATE)
+    x = shap.sample(x, min(60, len(x)), random_state=StaticValue.RANDOM_STATE)
     explainer = shap.KernelExplainer(model.predict, x, feature_names=feature_names)
     shap_values = explainer(x)
 

data/notes.md

@@ -10,6 +10,8 @@
 -  ଘ(੭ˊ꒳​ˋ)੭
 
 ## 注意事项
+- 程序部署在Hugginface上，为gradio框架，由于gradio直接部署的方案是刚出的，所以有时候程序可能存在不稳定
+- 若程序出现ERROR，刷新重新开始即可
 - 模型训练和可视化过程暂未实现进度条，后续版本可能会出该功能
 
 ## 解释