TensorFlow.js学习笔记—曲线拟合

这个DEMO是参考了TensorFlow.js官网的教程，融合了一些自己的实现。

具体效果是怎么样的呢？少废话，先看东西：

这是一个二次函数曲线的拟合，效果我个人觉得还可以。

为了实现这个效果，首先我们得有要有数据，不然我们拟合什么？数据就是下图所示的这些散点。

为了巧妇能有米来炊，我们首先得要实现一个产生数据的函数。

function generateData(numPoints, coeff, sigma = 0.5) {
   
  return tf.tidy(() => {
   
    const [a, b, c] = [
      tf.scalar(coeff.a),
      tf.scalar(coeff.b),
      tf.scalar(coeff.c)
    ];
  
    const xs = tf.randomUniform([numPoints], -4, 2);
    const ys = a.mul(xs.pow(tf.scalar(2, 'int32')))
      .add(b.mul(xs))
      .add(c)
      //添加噪声
      .add(tf.randomNormal(