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

Example to export model to ONNX #691

Open
wants to merge 4 commits into
base: main
Choose a base branch
from

Conversation

twoertwein
Copy link

Closes #689

I added an example to demonstrate that it is possible to convert an RSMTool model to ONNX.

I just implemented the transformations to make the conversion work for the model I was dealing with :) The example will error when it encounters skll/rsmtool features that it does not (yet) support/when the conversion fails.

I was dealing with a simple regression model. fast_predict takes 0.1s while the converted ONNX+numpy model takes only 0.00014s

@desilinguist
Copy link
Member

This is great @twoertwein! We will also need to add details about this to the documentation explaining the use case and the dependencies needed (e.g, skl2onnx etc.).

@twoertwein
Copy link
Author

We will also need to add details about this to the documentation explaining the use case and the dependencies needed (e.g, skl2onnx etc.).

In the form of a quick tutorial like this one https://github.com/EducationalTestingService/rsmtool/blob/main/doc/tutorial_rsmcompare.rst.inc or should this be documented differently?

@desilinguist
Copy link
Member

A tutorial would be fantastic!

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

Slim installation option or model inference export option
2 participants