在javascript中加载经过训练的机器学习模型

问题描述

我正在python中使用sklearn来构建/训练某些模型（Random Forest Regressor，Kmeans，SVM等），我想在webapp Javascript / html中使用那些训练有素的模型。有办法吗？

我已经看到tensorflow.js允许使用keras模型进行此类操作。但是在我看来，Python中的TF / keras仅限于神经网络。

我也见过sklearn-porter，但它似乎仅限于某些特定模型（主要是分类）。如果有人设法使用它，他还能告诉我更多吗？

在此先感谢您的帮助

解决方法

作为一种选择，您可以简单地构建RESTfull API in Python（例如，使用Flask）。轻松之后，只需从Javascript请求预测即可。

这种方法没有什么好处：

• 除了模型之外，您可能还需要数据准备功能；
• 您必须分离项目；
• 使用REST API，您可以轻松添加模型的新版本；