jpmml / jpmml-xgboost

Java library and command-line application for converting XGBoost models to PMML
GNU Affero General Public License v3.0
128 stars 43 forks source link

'binary:hinge' are not supported #40

Closed Ying456123 closed 4 years ago

Ying456123 commented 5 years ago

@vruusmann I'm converting a xgboost model to a PMML model, and find it is still not supported by jpmml-xgboost. Could you help to support it when you have time? Very appreciated. The model:

import xgboost as xgb
import numpy as np
data = np.random.rand(5, 10)  # 5 entities, each contains 10 features
label = np.random.randint(2, size=5)  # binary target
dtrain = xgb.DMatrix(data, label=label)
param = {'max_depth': 2, 'eta': 1, 'silent': 1, 'objective': 'binary:hinge'}
plst = param.items()
model = xgb.train(plst, dtrain)
tuanvayn commented 4 years ago

did you resolve it? I've changed by adding objective case like below but more error thrown :( image

tuanvayn commented 4 years ago

Hi. Have someone solved?