LogisticRegression model to TFLite?

prateekg · 01-09-2023 07:58 PM

Hi All,

I need your guidance!

I have trained a logistic regression model from the scikit-learn library and saved the model as .cpickle file.

But the goal is to use this model in an Android app so my android team needs the model in supportive format. I am not sure if .cpickle is supportive for them or not.

So my query is in which format I can save the model so that mobile team can easily integrate it with the Android app. I know about the TFLite format but how can I convert the model to this format or is there any other way?

Please share your thoughts on this.

anjelab

This is more of a Tensorflow question and it would better help you to participate in their Community as well. I have found though this SO post that could be helpful on conversion to tflite.

sascha_heyer

here is a great article that describes on how to transform your model into ONNX format which can be executed on android devices.
https://towardsdatascience.com/deploying-scikit-learn-models-in-android-apps-with-onnx-b3adabe16bab