受限制的玻尔兹曼机器

时间:2016-09-19 04:56:16

标签: r machine-learning deep-learning rbm

我目前正在尝试使用deepnet包在R中使用RBM。我使用自己的数据集训练了一个RBM,有3个输入点。在训练网络后,我得到了2组权重和2组偏差。 我的代码像这样运行

a<-matrix(c(1,0,0,0,1,0,0,0,1,1,1,1),nrow=4,ncol=3,byrow=T)
RBM_trn<-rbm.train(a, 2, numepochs = 30, batchsize = 100, learningrate=0.8,
momentum =0.5 ,visible_type = "bin",hidden_type = "bin" , cd = 1)
RBM_trn

我得到的结果是2.我得到两个2x3权重矩阵。另一个矩阵是什么意思?

1 个答案:

答案 0 :(得分:1)

请检查:https://github.com/cran/deepnet/blob/master/R/rbm_train.R

其中W和B对应于每次迭代时学习的重量和偏差,使用随机(或小批量)梯度下降来优化成本函数,VW和VB也结合了动量(有助于最小化噪声重量更新)。 / p>