基于TensorFlow.js的JavaScript机器学习( 十 )

单位(链接)设置权重矩阵在层中的大小 。 在这里将其设置为1 , 我们可以说每个数据输入特性都有一个权重 。

model.add(tf.layers.dense({units: 1));

上面的代码创建了我们的输出层 。 我们将单位设置为1 , 因为我们要输出1这个数字 。

创建实例

将以下代码添加到前面定义的运行函数中 。

这样可以创建实例模型 , 并且在网页上有显示层的摘要 。

步骤4:为创建准备数据

为了获得TensorFlow.js的性能优势 , 使培训机器学习模型实用化 , 我们需要将数据转换为Tensors 。

将以下代码添加到script.js文件中 。

接下来 , 我们可以分析一下将会出现什么情况 。

推荐阅读