rlab/dnn/examples/crbm.js

Javascript scientific library like R

/**
 * Created by joonkukang on 2014. 1. 15..
 */
var dnn = require('../lib/dnn');
var data = [[0.4, 0.5, 0.5, 0.,  0.,  0.7],
    [0.5, 0.3,  0.5, 0.,  1,  0.6],
    [0.4, 0.5, 0.5, 0.,  1,  0.9],
    [0.,  0.,  0., 0.3, 0.5, 0.],
    [0.,  0.,  0., 0.4, 0.5, 0.],
    [0.,  0.,  0., 0.5, 0.5, 0.]];

var crbm = new dnn.CRBM({
    input : data,
    n_visible : 6,
    n_hidden : 5
});

crbm.set('log level',1);

crbm.train({
    lr : 0.6,
    k : 1,
    epochs : 1500
});



var v = [[0.5, 0.5, 0., 0., 0., 0.],
    [0., 0., 0., 0.5, 0.5, 0.]];

console.log(crbm.reconstruct(v));
console.log(crbm.sampleHgivenV(v)[0]);