ONNX export doesn't support categorical features

[wboleksii]

Hi,

Is there a plan to support categorical features when exporting to Onnx format?

[annaveronika]

It would be great to add it, I'm adding a help wanted tag, because we will not be able to implement it in the close future. But it is easy to do, examples can be found in CoreML and PMML support for categorical features. If you are interested, you could try implementing this one.

[chebbyChefNEQ]

Hi, I'd love to have this feature and contribute some code. Could you point me to the necessary components and docs? @annaveronika

Thank you!

[chebbyChefNEQ]

Looks like to support categorical ONNX export the code need some ai.onnx.ml.CategoryMapper nodes? Once all the features are mapped to ints we can just use the AddTree path.

Is my understanding correct?

[alexeymaksakov-tomtom]

Any updates on this one?

[mirecl]

Any updates on this one?

[Evgueni-Petrov-aka-espetrov]

hi @mirecl !
thanks for asking 👍
so far, nobody has volunteered to implement this feature 😞
i think that the easiest thing is to support one-hot encoded categorical features

[alexeymaksakov-tomtom]

What if one-hot encoding results in 200+ features? :)