{
const config = {
binaryThresh: 0.5,
hiddenLayers: [3],
activation: 'sigmoid',
leakyReluAlpha: 0.01,
};
const net = new brain.NeuralNetwork(config);
const red = [255, 0, 0];
const green = [0, 255, 0];
const blue = [0, 0, 255];
const yellow = [255, 255, 0];
const orange = [255, 128, 0];
function normiliseData(colour) {
return colour.map((c) => c / 255);
}
console.log(normiliseData(orange));
net.train([
{ input: normiliseData(red), output: normiliseData(blue) },
{ input: normiliseData(blue), output: normiliseData(red) },
{ input: normiliseData(green), output: normiliseData(yellow) },
{ input: normiliseData(yellow), output: normiliseData(orange) },
], { log: true });
return [
net.run(normiliseData(red)),
net.run(normiliseData(blue)),
net.run(normiliseData(green)),
net.run(normiliseData(yellow)),
];
}