我有一个朋友使用Python和TensorFlow构建机器学习模型。我想在我的网站上使用它,我的网站后端使用nodejs。
有没有办法将TensorFlow构建和训练的模型整合到JavaScript代码中,这样我就可以在我的网站上提供基于机器学习的服务?
回答:
可以看看TensorFlow.js,这是一个专门用于TensorFlow模型的基于浏览器的JavaScript库。
更具体地说,很可能你可以使用tf.loadModel()
直接将你朋友的模型加载到你的网站上,然后在那里运行它。
有一个教程关于将Keras模型导入到TensorFlow.js。希望你的朋友知道什么是Keras,并且知道如何用Keras生成他的模型。
如果是直接使用TensorFlow,那么请按照这个指南将他的模型转换为TensorFlow.js。