SHAP- TreeExplainer (модель) - какое это поле ожидаемого_значения? - PullRequest
0 голосов
/ 20 февраля 2020

Я использовал SHAP, чтобы объяснить мой RF

RF_best_parameters = RandomForestRegressor(random_state=24, n_estimators=100)
RF_best_parameters.fit(X_train, y_train.values.ravel())
shap_explainer_model = shap.TreeExplainer(RF_best_parameters)

Класс TreeExplainer имеет атрибут, который называется - expected_value. Мое первое предположение, что это поле является средним значением предсказанного y, согласно X_train (я также читал это здесь )

Но это не так. Вывод команды:

shap_explainer_model.expected_value

равен 0,2381. Вывод команды:

RF_best_parameters.predict(X_train).mean()

равен 0,2389. Как мы видим, значения не совпадают. Так в чем же значение ожидаемого_значения? Спасибо.

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...