rsmtool icon indicating copy to clipboard operation
rsmtool copied to clipboard

Example to export model to ONNX

Open twoertwein opened this issue 1 year ago • 3 comments

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

twoertwein avatar Jan 02 '25 14:01 twoertwein