如何在Android上部署scikit学习模型?

时间:2019-05-16 09:37:54

标签: android tensorflow machine-learning keras scikit-learn

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

我是否使用Tensorflow重新实现随机森林?我还能怎么做才能将其部署到android?

我不想在android上训练我的模型,但是在其上使用我的自定义训练模型。

1 个答案:

答案 0 :(得分:0)

您可以使用m2cgen(模型2代码生成器),这是一个轻量级的库,可以提供一种简便的方法将经过训练的统计模型转换为本地代码(Python,C,Java,Go,JavaScript,Visual Basic,C#, PowerShell,R,PHP,Dart,Haskell,Ruby,F#)