auto-sklearn icon indicating copy to clipboard operation
auto-sklearn copied to clipboard

Published 20 hours ago • verified publisher icon automl

Allow autosklearn to export ONNX model

Open shanalikhan opened this issue 3 years ago • 10 comments

Onnx allows to convert the sk-learn model object to ONNX object.

Is there any way to convert the autosklearn object to ONNX object or any way to get the model sklearn classifier object from auto-sklearn to convert to onnx.

References:

https://github.com/onnx/sklearn-onnx http://onnx.ai/sklearn-onnx/index.html

shanalikhan avatar Nov 13 '20 14:11 shanalikhan

This issue has been automatically marked as stale because it has not had recent activity. It will be closed if no further activity occurs for the next 7 days. Thank you for your contributions.

github-actions[bot] avatar May 05 '21 01:05 github-actions[bot]

Trying to re-up this thread. I agree with @shanalikhan : a way of serializing an AutoSKL model into ONNX would be great. Would it be possible to reopen this issue ?

francoisWeber avatar Jun 16 '22 10:06 francoisWeber

Hi @francoisWeber, we will consider sklearn model export as part of the next milestone v0.16. There was an initial attempt but as far as I remember, there is some structural issues regarding that. I don't think we will directly support ONNX support for ASKL but hopefully sklearn model export should be enough?

eddiebergman avatar Jun 16 '22 11:06 eddiebergman

Hi @eddiebergman and thanks for this quick answer. ASKL to SKL conversion is a really good news. IMHO as soon as a pure-SKL object is obtained, sklearn-onnx will be enough to convert the whole pipeline into an ONNX file (as pointed by @shanalikhan ). This pure-SKL export is a great news for industrial considerations :) Thanks ! Looking forward to play with v0.16 !

francoisWeber avatar Jun 16 '22 12:06 francoisWeber

No promises but if sklearn exports make the milestone list of things to do then I'll update this issue :)

eddiebergman avatar Jun 16 '22 13:06 eddiebergman

Hey @eddiebergman , May I know

  1. if sklearn export has made it to the milestone of things to be done in v0.16?
  2. any estimated timeline for when it (v0.16) could be ready?

arkarn avatar Sep 14 '22 07:09 arkarn

Hi @arkarn,

Sorry, we've currently become quite busy with ICLR deadlines so it's been delayed. We're currently benchmarking the v0.15 changes so once that's merged we can begin proper work on v0.16.

ICLR deadlines are 28th Sept so you can expect some v0.16 work to begin in October with the release being in November sometime. We've already begun a few PR's for v0.16 though so it's not from scratch :)

Sorry for the delays, Best, Eddie

eddiebergman avatar Sep 14 '22 10:09 eddiebergman

HI @eddiebergman,

I found v0.16dev on this website (https://automl.github.io/auto-sklearn/development/manual.html). I'm curious about whether autosklearn can export ONNX model in v0.16dev? If not, may I know any estimated timeline for when it could be ready?

Thanks.

jmren168 avatar Feb 23 '23 13:02 jmren168

Hi @eddiebergman,

Sorry to bother you again and again. May I know do you have any estimated timeline for when supporting of onnx ouput could be ready?

Many thanks.

jmren168 avatar May 22 '23 03:05 jmren168

Hey, are there any news on the ASKL -> ONNX export or ASKL -> SKL (-> ONNX)? We would very much like to use the models obtained via ASKL but without the export we have no way to deploy them.

If the full export is difficult, could you perhaps add something like autosklearn.classification.AutoSklearnClassifier_Exportable() that would only use exportable components?

Thanks.

otakar-trunda avatar Jun 27 '23 09:06 otakar-trunda