Published
Edited
Apr 13, 2022
Insert cell
Insert cell
{
const trainingData = [
{ input: [0, 1], output: [1] },
{ input: [0, 0], output: [0] },
{ input: [1, 1], output: [0] },
{ input: [1, 0], output: [1] },
];
const net = new brain.NeuralNetwork({
// hiddenLayers: [12, 8],
});
console.time("CPU training");
const output = net.train(trainingData, {
iterations: 50000, // maximum training iterations
log: true, // console.log() progress periodically
logPeriod: 10000, // number of iterations between logging
});
console.timeEnd("CPU training");
console.log(`CPU training output`, output);
console.time("GPU training");
const netGPU = new brain.NeuralNetworkGPU({
// hiddenLayers: [12, 8],
});
const outputGPU = netGPU.train(trainingData, {
iterations: 50000, // maximum training iterations
log: true, // console.log() progress periodically
logPeriod: 10000, // number of iterations between loggin
});

return { output, outputGPU };
}
Insert cell
Insert cell
Insert cell

Purpose-built for displays of data

Observable is your go-to platform for exploring data and creating expressive data visualizations. Use reactive JavaScript notebooks for prototyping and a collaborative canvas for visual data exploration and dashboard creation.
Learn more