Загрузите и используйте модель XGBoost PMML или XGBoost .rds в python sklearn без потери своих зависимостей / характера

Question

У меня есть модель XGB, которая встроена в R, я хотел использовать ту же модель в Python для предсказаний без потери ее природы / встроенных зависимостей.

Я пробовал rpy2 и sklearn-pmml-model, похоже, что для моделей XGBoost нет никакой помощи.

Ценю любую помощь.

Answer 1

Вы можете использовать PyPMML для оценки XGBoost PMML в Python, например:

from pypmml import Model

model = Model.fromFile('the/pmml/model/file/path')
result = predict(data)

data может быть в dict, Series или DataFrame of Pandas или в строке JSON.