如何将 scikit-learn 模型部署到 android？

Posted 2023-03-12

【中文标题】如何将 scikit-learn 模型部署到 android？

【英文标题】：How do I deploy a scikit-learn model to android?

【发布时间】：2019-10-03 13:10:18

【问题描述】：

我有一个实现 sklearn 的 RandomForestClassifier 的模型。我已将模型保存在“pickle”文件中，我想将其部署到 android 应用程序中。 我已经看到这可以使用 Tensorflow Lite 文件来完成，并且可以将“hdf5”文件转换为相同的文件。但是我不知道如何将我的泡菜文件转换为 hdf5，然后再转换为 tflite。

我是否使用 Tensorflow 重新实现随机森林？我还能做些什么来将其部署到 android？

我不想在 android 上训练我的模型，而是使用我的自定义训练模型。

【问题讨论】：

Using sklearn in android device的可能重复

【参考方案1】：

您可以使用 m2cgen（Model 2 代码生成器），它是一个轻量级库，可以提供一种简单的方法将经过训练的统计模型转换为本机代码（Python、C、Java、Go、javascript、Visual Basic、C#、 PowerShell、R、php、Dart、Haskell、Ruby、F#)