- added one hot gen

git-svn-id: http://moon:8086/svn/software/trunk/projects/Rbm@841 b431acfa-c32f-4a4a-93f1-934dc6c82436
This commit is contained in:
2022-01-19 11:02:11 +00:00
parent 963ee3cd9d
commit 7232870123
3 changed files with 43 additions and 26 deletions
+1 -1
View File
@@ -77,7 +77,7 @@ void RnnStack::train(const arma::mat& batch, Rbm::IListener* pListener)
arma::mat batch_shifted = batch_padded.rows(k, numTraining+k-1);
arma::mat vc = v_to_vc(batch_shifted, c);
pLayer->train(vc, pListener);
c = pLayer->to_h_gibbs(vc);
pLayer->gibbs_vh(vc, c);
pLayer = pLayer->next;
}
}