Tensorflow.js tf.data.Dataset.filter()函数

Tensorflow.js 是谷歌开发的一个开源库，用于在浏览器或节点环境中运行机器学习模型和深度学习神经网络。它还可以帮助开发人员用 JavaScript 语言开发 ML 模型，并且可以直接在浏览器或 Node.js 中使用 ML。

tf.data.Dataset.filter()函数用于根据谓词过滤数据集。

句法：

filter(predicate)

参数：

• predicate ：将数据集元素映射到布尔值或 Promise 的函数。

返回值：它返回 tf.data.Dataset

示例 1：

Javascript

const gfg = tf.data.array([8, 12, 25, 6, 34, 2, 67, 43])
   .filter(x => x >= 10);
await gfg.forEachAsync(geeks => console.log(geeks));

输出：

12
25
34
67
43

示例 2：

Javascript

const gfg = tf.data.array([10, 21, 36, 24, 54, 65, 72, 90, 95, 12])
   .filter(x => x%6 === 0);
await gfg.forEachAsync(geeks => console.log(geeks));

输出：

36
24
54
72
90
12

参考： https://js.tensorflow.org/api/latest/#tf.data.Dataset.filter