У меня есть модель XGBoost, обученная в Sagemaker и загруженная на моем локальном компьютере. Когда вы распаковываете файл tar.gz, вы получаете файл объекта модели (бустер). Я хочу построить графики частичной зависимости для некоторых функций моего набора данных (пакетное преобразование с прогнозами) с использованием экземпляра модели Sagemaker. Однако я получаю следующую ошибку.
ValueError: «оценка» должна быть подходящим регрессором или классификатором.
Код, который я использую для построения графика как показано ниже-
from sklearn.inspection import partial_dependence
pardep = partial_dependence(model, X, 'num_org')
Здесь модель X - это фрейм данных пакетного преобразования с ~ 100 столбцами, а 'num_org' - это то, что я хочу построить в отношении прогнозов.