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

将以下函数添加到script.js文件中以定义模型 。

这是我们可以在tensorflow.js中定义的最简单的模型之一 , 我们来试下简单分解每一行 。

实例化模型

const model = tf.sequential();

这将实例化一个tf.model对象 。 这个模型是连续的 , 因为它的输入直接流向它的输出 。 其他类型的模型可以有分支 , 甚至可以有多个输入和输出 , 但在许多情况下 , 你的模型是连续的 。

添加层

model.add(tf.layers.dense({inputShape: [1
units: 1 useBias: true));

这为我们的网络添加了一个隐藏层 。 因为这是网络的第一层 , 所以我们需要定义我们的输入形状 。 输入形状是[1
, 因为我们有1这个数字作为输入(给定房间的房间数) 。

推荐阅读