Блог

Я пытаюсь проверить важность функции модели, которую я создаю с помощью SHAP.

Он отлично работает, используя следующее, но я хотел бы иметь график только с подмножеством функций. Возможно ли это?

 shap.summary_plot(shap_values, features=X_train2, feature_names=X_train2.columns, max_display=30)
  

Я попытался определить подмножество функций и feature_names с помощью iloc в наборе данных, например features=X_train.iloc[;23:] , но это не работает.

Вам нужно будет подмножить как функции, так и значения shap_values, иначе они не будут выстраиваться в линию. Для чего это стоит, новый SHAP API, который использует объекты shap.Explanation, упростит это, поскольку он позволяет одновременно выполнять простое параллельное срезание функций и значений SHAP.