https://www.tensorflow.org/js/tutorials/training/linear_regression