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

Question

我在 python 中使用 sklearn 来构建/训练一些模型（随机森林回归器、Kmeans、SVM、...），并且我想在 webapp Javascript/html 中使用这些训练过的模型。有办法吗？

我已经看到 tensorflow.js 允许使用 keras 模型做这些事情。但在我看来，python 中的 TF/keras 仅限于神经网络。

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

提前感谢您的帮助

Answer 1

作为一个选项，您可以简单地构建RESTfull API in Python（例如使用 Flask）。在这很容易之后，只需从 Javascript 请求预测。

这种方法的好处很少：

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