Published
Edited
Jan 11, 2021
Insert cell
md`# Brain.js Issue #682`
Insert cell
Insert cell
{
//Using GPU here will be very expensive, because input is so tiny. Using CPU instead.
//const net = new brain.NeuralNetworkGPU();
const net = new brain.NeuralNetwork();
net.train([
{
input: [0],
output: [3],
},
{
input: [1],
output: [3],
},
{
input: [2],
output: [3],
},
{
input: [3],
output: [3],
},
{
input: [4],
output: [3],
},
{
input: [5],
output: [1],
},
{
input: [6],
output: [3],
},
{
input: [7],
output: [2],
},
{
input: [8],
output: [1],
},
{
input: [9],
output: [1],
}], { iterations: 1000, log: true });
const jsonString = JSON.stringify(net.toJSON());
const newNet = new brain.NeuralNetwork().fromJSON(JSON.parse(jsonString));
return newNet.run([8]);
}
Insert cell
Insert cell
{
const net = new brain.recurrent.LSTM({ hiddenLayers: [3] });
net.train([
{
input: [0],
output: [3],
},
{
input: [1],
output: [3],
},
{
input: [2],
output: [3],
},
{
input: [3],
output: [3],
},
{
input: [4],
output: [3],
},
{
input: [5],
output: [1],
},
{
input: [6],
output: [3],
},
{
input: [7],
output: [2],
},
{
input: [8],
output: [1],
},
{
input: [9],
output: [1],
}], { iterations: 4000, log: true });
const jsonString = JSON.stringify(net.toJSON());
const newNet = new brain.recurrent.LSTM();
newNet.fromJSON(JSON.parse(jsonString));
return newNet.run([8]);
}
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