TensorFlow.js: Train Multivariate Regression with the Layers API