Polynomial Regression with TensorFlow.js