index.js

const tf = require("@tensorflow/tfjs-node");

(async () => {
  const model = tf.sequential({
    layers: [tf.layers.dense({ inputShape: [1], units: 1 })]
  });

  model.compile({ optimizer: "sgd", loss: "meanSquaredError" });

  // y = 2X - 1
  const xs = tf.tensor([-1.0, 0.0, 1.0, 2.0, 3.0, 4.0]);
  const ys = tf.tensor([-3.0, -1.0, 1.0, 3.0, 5.0, 7.0]);

  await model.fit(xs, ys, { epochs: 1000 });

  model.predict(tf.tensor([10.0])).print();
})();

package.json

{
  "main": "index.js",
  "dependencies": {
    "@tensorflow/tfjs-node": "^1.2.8"
  }
}