用XgBoostClassifier预测Xgboost DMatrix对象

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方法？我建议您先在本地调试它，因为它似乎不是专门针对传奇人物的。