| 1 | var R = require('../rlab');
|
| 2 |
|
| 3 | var RBM = R.NN.RBM;
|
| 4 |
|
| 5 | var data = [
|
| 6 | [1,1,1,0,0,0],
|
| 7 | [1,0,1,0,0,0],
|
| 8 | [1,1,1,0,0,0],
|
| 9 | [0,0,1,1,1,0],
|
| 10 | [0,0,1,1,0,0],
|
| 11 | [0,0,1,1,1,0]];
|
| 12 |
|
| 13 | var rbm = new RBM(R, {
|
| 14 | input : data,
|
| 15 | nVisible : 6,
|
| 16 | nHidden : 2
|
| 17 | });
|
| 18 |
|
| 19 | rbm.train({
|
| 20 | lr : 0.6,
|
| 21 | k : 1,
|
| 22 | epochs : 500
|
| 23 | });
|
| 24 |
|
| 25 | var v = [[1, 1, 0, 0, 0, 0],
|
| 26 | [0, 0, 0, 1, 1, 0]];
|
| 27 |
|
| 28 | console.log(rbm.reconstruct(v).strM());
|
| 29 | console.log(rbm.sampleHgivenV(v)[0].strM()); |
| \ | No newline at end of file |