TensorFlow.js: Fitting a curve to synthetic data

Original Data (Synthetic)
True coeffecients:
Fit curve with random coefficients (before training)
Random coeffecients:
Fit curve with learned coefficients (after training)
Learned coeffecients: