Tensorflow.js 简介

什么是 Tensorflow.js？

TensorFlow.js 是一个 JavaScript 库，用于在 Web 应用程序和 Node.js 中训练和部署机器学习模型。您可以使用 tensorflow.js 从头开始开发机器学习模型，也可以使用提供的 API 在浏览器或 Node.js 服务器上训练现有模型。

要了解更多信息，您可以直接访问此链接：https://www.tensorflow.org/resources/learn-ml/basics-of-tensorflow-for-js-development。

先决条件：在启动 Tensorflow.js 之前，您需要了解以下内容：

对于浏览器：

• HTML：需要 HTML 的基础知识
• JavaScript：需要良好的 JS 知识

对于服务器端：

• Node.js：熟练掌握 Node.js。此外，由于 Node.js 是一个 JS 运行时，因此对 JavaScript 进行命令会很有帮助。

其他需求：

• NPM 或 Yarn：这些是需要安装在系统中的软件包。

设置 Tensorflow.js：

浏览器设置在基于浏览器的应用程序中添加 TensorFlow.js 有两种方法：

• 使用脚本标签。
• 从 NPM 安装

1. 使用脚本标签：将以下脚本标签添加到您的主 HTML 文件中。

例子：

HTML

  

    
    

  

    
  
        // Value of a scalar
        var value = "Hello Geeks! Tensorflow here."
  
        // Creating the value of a scalar
        var tens = tf.scalar(value)
  
        document.write(tens)
    

  

输出：

2. 使用 NPM/yarn：我们可以使用 NPM 或 Yarn 来安装 tensorflow.js。

yarn add @tensorflow/tfjs

要么

npm install @tensorflow/tfjs

Node.js 设置：

选项 1：安装带有原生 C++ 绑定的 TensorFlow.js。

yarn add @tensorflow/tfjs-node

要么

npm install @tensorflow/tfjs-node

选项 2：（仅限 Linux）如果您的系统具有支持 CUDA 的 NVIDIA® GPU，则使用 GPU 包甚至可以获得更高的性能。

yarn add @tensorflow/tfjs-node-gpu

要么

npm install @tensorflow/tfjs-node-gpu

选项 3：安装纯 JavaScript 版本。这是性能方面最慢的选项。

yarn add @tensorflow/tfjs

要么

npm install @tensorflow/tfjs