Предсказания для объекта Xgboost DMatrix с помощью XgBoostClassifier

Question

Я работаю с предварительно обученной моделью на AWS SageMaker,

Мне нужно предоставить predict_fn, который будет делать реальный прогноз.

def predict_fn(request_body, model):
    prediction = model.predict(request_body)
    return prediction

Это возвращает ошибку: TypeError: no supported conversion for types: (dtype(‘O’),)

модель является объектом XGboostClassifier, requestbody является объектом xgboost.core.DMatrix

Answer 1

Вы также реализовали свои собственные методы model_fn и input_fn? Я бы посоветовал вам сначала попробовать отладить это локально, так как похоже, что это не спецификатор sagemaker c.