Tensorflow.js - how to create a model for a simple sum

Question

I am starting studies with tensoflow.js. I\'ve seen and recreated the prediction example for a linear function; where you initially have:

async function learn