-
Notifications
You must be signed in to change notification settings - Fork 59
Expand file tree
/
Copy path01_basic.js
More file actions
25 lines (20 loc) · 738 Bytes
/
01_basic.js
File metadata and controls
25 lines (20 loc) · 738 Bytes
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
const brain = require('brain.js');
const network = new brain.NeuralNetwork();
// network.train([
// { input: [0, 0, 0], output: [0] },
// { input: [0, 0, 1], output: [0] },
// { input: [0, 1, 1], output: [0] },
// { input: [1, 0, 1], output: [1] },
// { input: [1, 1, 1], output: [1] }
// ]);
network.train([
{ input: [1, 2], output: [1] }, // Team 2 wins
{ input: [1, 3], output: [1] }, // Team 3 wins
{ input: [2, 3], output: [0] }, // Team 2 wins
{ input: [2, 4], output: [1] }, // Team 4 wins
{ input: [1, 2], output: [0] }, // Team 1 wins
{ input: [1, 3], output: [0] }, // Team 1 wins
{ input: [3, 4], output: [0] } // Team 3 wins
]);
const output = network.run([1, 4]);
console.log(`Prob: ${output}`);