-
Notifications
You must be signed in to change notification settings - Fork 125
Open
Description
Hello there,
I was trying to see how I would write a custom parser to make it so that a model converted from xgboost classifier could only return probabilities, not the standard label and dictionary of probabilities. I eventually realized it didn't matter what I passed to custom_parsers in convert_sklearn, it returned the same output.
Diving into the code I noticed that, unless I'm missing something, this custom_parsers argument is unused in the parse_sklearn_simple_function.
Is there intention to make use of this parameter? Should it be removed or perhaps an error thrown that custom parsers are currently unsupported?
sklearn-onnx/skl2onnx/_parse.py
Line 104 in 99939ef
| def _parse_sklearn_simple_model(scope, model, inputs, custom_parsers=None, alias=None): |
Reactions are currently unavailable
Metadata
Metadata
Assignees
Labels
No labels