12、开启 TensorFlow.js 之旅与深度学习 API 解析

开启 TensorFlow.js 之旅与深度学习 API 解析

一、TensorFlow.js 模型创建与训练

1. 数据处理
   • 首先，将存储在 iris 变量中的数据读取到 tensor2d 对象。在 index.js 文件中添加如下代码：

const trainingData = tf.tensor2d(iris.map(item=> [
    item.sepal_length, item.sepal_width, item.petal_length,
    item.petal_width
]),[144,4])

- 接着，对可能的三种花卉品种进行独热编码：

const outputData = tf.tensor2d(iris.map(item => [
    item.species === 'setosa' ? 1 : 0,
    item.species === 'virginica' ? 1 : 0,
    item.species === 'versicolor' ? 1 : 0
]), [144,3])

2. 模型构建