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

epochs指的是模型将查看你提供的整个数据集的次数 。 在这里 , 我们通过数据集进行50次迭代 。

启动列车环路

model.fit是我们调用的启动循环的函数 。 它是一个异步函数 , 因此我们返回它给我们的特定值 , 以便调用者可以确定运行结束时间 。

为了监控运行进度 , 我们将一些回调传递给model.fit 。 我们使用tfvis.show.fitcallbacks生成函数 , 这些函数可以为前面指定的“损失”和“毫秒”度量绘制图表 。

把它们放在一起

现在我们必须调用从运行函数定义的函数 。

将以下代码添加到运行函数的底部 。

刷新页面时 , 几秒钟后 , 你应该会看到图形正在更新 。

这些是由我们之前创建的回调创建的 。 它们在每个时代结束时显示丢失(在最近的批处理上)和毫秒(在整个数据集上) 。

推荐阅读