Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[Feature Request] Allow splitting the created node graph, just before the model runs, so we get the preprocessed data and use them #22621

Open
alarv opened this issue Oct 28, 2024 · 0 comments
Labels
feature request request for unsupported feature or enhancement

Comments

@alarv
Copy link

alarv commented Oct 28, 2024

Describe the feature request

As a modeler, I would like ONNX to provide me with the ability to split the node graph for my scikit-learn models. I see most scikit-learn models are stored as a node graph of preprocessing steps and the last step is the actual model.

Describe scenario use case

I want to calculate the domain of applicability for the input data to the model and I can't somehow split the created node graph cleanly, so I can get the preprocessed data just before they enter the model.

The issue is that this (the model being the last node in the created node graph) is not the case for all the scikit-learn models and I can't safely extract the preprocessed data just before the enter the model node.

It'd be nice if onnx's model was somehow split into preprocessing nodes, model nodes and post-processing nodes.

@alarv alarv added the feature request request for unsupported feature or enhancement label Oct 28, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
feature request request for unsupported feature or enhancement
Projects
None yet
Development

No branches or pull requests

1 participant