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

时间:2020-08-13 13:04:36

标签: javascript python machine-learning scikit-learn tensorflow.js

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

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

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

在此先感谢您的帮助

1 个答案:

答案 0 :(得分:0)

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

这种方法没有什么好处:

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