将 tensorflow 与 nodejs 库一起用于机器学习模型

Posted 2023-02-23

【中文标题】将 tensorflow 与 nodejs 库一起用于机器学习模型

【英文标题】：Using tensorflow with nodejs library for machine learning models

【发布时间】：2020-03-05 14:07:22

【问题描述】：

我已阅读到不同的库可用于 Tensorflow。 （在 Python、C++、javascript 中）。

“Tensorflow.js 是一个 JavaScript 库，用于在 Node.js 中训练和部署机器学习模型”。 那么这是否意味着 Tensorflow 也支持 Nodejs？

我对 js/javascript/nodejs 不太了解。

谢谢， 马扬克

【参考方案1】：

没错。看这里 - https://www.npmjs.com/package/@tensorflow/tfjs

它甚至可以在浏览器中使用。请参阅上面链接中的Develop ML in the Browser 部分。

【讨论】：

感谢 Ashish，所以，我可以将 nodejs 用于 tensorflow，但在 tensorflow 官方网站 (tensorflow.org/lite/guide/inference) 上，我可以在“支持的平台 > Linux”部分看到，用于在 TF lite 模型上运行推理可用的 API 仅在 Python 和 C++ 中。那么，节点不支持运行推理吗？基本上我的最终目标是为 Linux 平台开发 Tensorflow。

【参考方案2】：

我找到的文档here 说这是可能的。

检查import * as tf from '@tensorflow/tfjs-node'

【讨论】：

我们可以使用基于 Nodejs 的库来实现 Tensorflow。仍然经过一些研究，我发现最好使用 Python/C++。这两个库有在线支持以及文档化的东西/帮助可用于帮助。