-
Notifications
You must be signed in to change notification settings - Fork 0
/
perceptron.js
56 lines (36 loc) · 1.26 KB
/
perceptron.js
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
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
'use strict';
const Stats = require('./utils/stats');
class Perceptron {
constructor(weights, activationFun) {
this.weights = weights;
this.fun = activationFun;
this.bias = 0;
this.r = 0.1;
}
train(trainingSet) {
console.time('Training');
for (let i = 0; i < trainingSet.length; i++) {
const inputs = trainingSet[i].inputs;
const desOutput = trainingSet[i].output;
const actOutput = this.fun(inputs, this.weights, this.bias);
for (let j = 0; j < this.weights.length; j++) {
this.weights[j] = this.weights[j] + this.r * (desOutput - actOutput) * inputs[j];
}
}
console.timeEnd('Training');
}
test(testSet) {
const stats = new Stats(testSet.length);
console.time('Testing');
for (let i = 0; i < testSet.length; i++) {
const testOutput = this.fun(testSet[i].inputs, this.weights, this.bias);
const isok = stats.isOk(testSet[i], [Number(testSet[i].inputs[1] > 0)], [testOutput]);
if (!isok) {
console.log(stats.lastFailedSample);
}
}
console.timeEnd('Testing');
stats.logSummary();
}
}
module.exports = Perceptron;