XGBRegressor converter is maybe broken

It seems there are problems with the XGBRegressor converter.

I’ve opened an issue at sklearn-onnx but they redirected me to here: https://github.com/onnx/sklearn-onnx/issues/321

Here is a repro: https://gist.github.com/victornoel/06d6231f6276719ddba53cb381dfd468

I casted an int column to str because in my original dataset, we have categories made of numbers so I thought this could be related…

As we can see, the differences are huge: the predictions should be between 0 and 1 and the differences can be as big as 1 in my tests!

>>> import test
>>> test.test()
[0.9019426  0.91082716 0.91868186 0.94926846 1.0898147 ]
min(Y)-max(Y): 0 1

[xadupre]

I was able to replicate. I’ll have a look tomorrow.

[victornoel]

@xadupre thank you, I created an issue there: https://github.com/microsoft/onnxruntime/issues/3144

Let me close the current issue then.