Как упростить работу с моделями машинного обучения
с помощью Shapash
Shapash работает с задачами регрессии, двоичной классификации или мультикласса. Он совместим со многими моделями:
Shapash основан на локальных вкладах, рассчитанных с помощью Shapley Value, Lime или любого другого метода, который позволяет вычислять суммируемые локальные вклады.
Вы можете установить пакет через pip:
$ pip install shapash
Сейчас мы рассмотрим, как можно использовать Shapash при работе с конкретной моделью. Для этого используем датасет со "стоимостью на жилье" от Kaggle, чтобы подобрать регрессор и спрогнозировать цены на недвижимость. Начнем с загрузки набора данных:
Обучаем, тестируем разделение и подгоняем модель: