Regression in Angular using TensorFlow.js