有没有办法在我的网站上使用TensorFlow模型？

Question

我有一个朋友使用Python和TensorFlow来构建机器学习模型。我想在我的具有nodejs后端的网站中使用它。

有没有办法将TensorFlow构建和训练的模型合并到JavaScript代码中，以便我可以在我的网站上提供基于机器学习的服务？

Answer 1

查看TensorFlow.js，它是TensorFlow模型的基于浏览器的JavaScript库。

更具体地说，您很可能可以使用tf.loadModel()将您朋友的模型直接加载到您的网站中，然后在那里运行。

importing a Keras model into TensorFlow.js有一个教程。希望你的朋友知道Keras是什么以及如何在Keras制作他的模型。

如果它是直接的TensorFlow，那么请按照this guide如何将他的模型转换为TensorFlow.js。