Skip to content

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
Collaborator

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
Collaborator

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