search

dswah / pyGAM

[HELP REQUESTED] Generalized Additive Models in Python
https://pygam.readthedocs.io
Apache License 2.0
852 stars 156 forks source link

Non-pickle export option? #344

Open nickeubank opened 4 months ago

nickeubank commented 4 months ago

Related to #308 , expansion of #201

It'd be great to have a non-pickle export option. As in #308, some people want to move models to other languages, but even for those who are staying in Python, an export option that isn't pickle-dependent would be really nice given how brittle pickles can be to version differences (having trouble not imaging a brittle pickle now...) and the fact some environments don't support their use for security reasons.

dswah commented 4 months ago

@nickeubank Interesting! What do you thinkwould be a good export format?

I see that scikit-learn supports pickle, joblib (similar to pickle), and onnx.

The Onnx approach would be cool, but requires some custom work. Maybe we can fork the sklearn-onnx repo to get a head start?