index.ts

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

require('@tensorflow/tfjs-node');

const model = tf.sequential();

const inputs = [
	[0,0],
	[0,1],
	[1,0],
	[1,1]
];

const targets = [
	[0],
	[1],
	[1],
	[1],
];

let intervID;

model.add(tf.layers.dense({
	units: 1,
	inputShape: [2],
	activation: 'sigmoid'
}));

// model.add(tf.layers.dense({
// 	units: 1,
// });

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

const x: tf.Tensor = tf.tensor(inputs);
const y: tf.Tensor = tf.tensor(targets);

const train = () => model.fit(x, y, {
	shuffle: true,
	epochs: 10
});

const trainModel = async ()=>{
	train()
	.then(result => {
		let loss = result.history.loss[0];
		console.log(loss)
		if(loss < 0.01){
			clearInterval(intervID);
			predict();
		}
	})
	.catch(console.error);
}

const predict = () => {
	model.predict(inputs).print();
}

setInterval(trainModel, 100);
// trainModel();