MetOffice / XBTs_classification

Project for the classification of eXpendable Bathy Thermographs
BSD 3-Clause "New" or "Revised" License
4 stars 2 forks source link

Save models in ONNX format #110

Open stevehadd opened 3 years ago

stevehadd commented 3 years ago

Currently we pickle the models to save the trained model state, which is unsatisfactory. there is now a scikit-learn add on called sklearn-onnx which allows us to save th classifier state in the ONNX format, which is (becoming?) the standard format for interoperability. https://onnx.ai/sklearn-onnx/supported.html