TensorFlow.js:擬合數據的曲線


已知数据大致符合函数 y = ax3 + bx2 + cx + d.(模型)

一,设置 a,b,c,d 为 tf 变量,在学习中需要一直改变

二,建立预测模型(predict),返回一个函数(返回 tf 操作数)

三,建立训练模块
  1. 定义损失函数 loss,返回一个 tf 数值,mean squared error (MSE)
  2. 定义一个优化器,tf 内置一些优化器,如 Stochastic Gradient Descent (SGD)
  3. 定义训练循环,循环中优化预测值(optimizer + predict + loss)

最后得到结果