导致shap慢的原因 shap不同版本差异

1.导致shap慢的原因

lightgbm如果把num_leaves设置很大的话,我这里设置的256,做CV会比较慢,用shap.TreeExplainer(model)会慢到合理时间内跑不出来,shap.TreeExplainer很受树复杂度的影响,shap版本0.37


2.shape不同版本的一些差异

shap不知道从哪个版本开始,我这里尝试了0.36和0.37,对于lightgbm二分类问题,shap_values = explainer.shap_values(X)输出的shap_values是一个list,list中包含两个数组ndarray,第一个是解释label 0的,第二个解释label 1的,两个数组的shape就等于X.shape,目前shape 0.39的API较0.37又有变化

留言

熱門文章