如何将 sklearn 模型嵌入移动设备以供离线使用?

How to embed sklearn models to mobile for offline use?

我需要在 iOS 和 Androis 应用程序中嵌入 scikit-learn 模型,特别是 svm 和随机森林。

我已经使用带有 tflite 的转换版本部署了 keras 模型,但找不到使用 sklearn 模型的方法。

有办法实现吗?

根据平台的不同,存在允许在移动设备上使用 Scikit-learn 的包装器库。

对于 iOS,您应该可以通过查看 coremltools.converters.sklearn.convert 使用 CoreML/CoreMLTools 来完成此操作。它确实提供 'random forest',目前不提供 svm,但基于库代码(即 coremltool + scikitlearn),您应该能够扩展它以使用 svm。

对于 Android,您也许可以找到有关 this question 的信息。