jens/Rbm
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

- assign training data with context

Browse Source 
- calculate context on demand if empty

git-svn-id: http://moon:8086/svn/software/trunk/projects/Rbm@773 b431acfa-c32f-4a4a-93f1-934dc6c82436
This commit is contained in:
jens
2022-01-10 12:19:48 +00:00
parent 9f73824bc4
commit 8c8e0b324d
5 changed files with 21 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files
source/RbmComponent.cpp
+3 -3
View File
@@ -279,12 +279,12 @@ void RbmComponent::gibbs(const arma::mat& vc)
arma::mat RbmComponent::getTraining() const
{
return to_vc(DrawVisibleTrain->getData(), DrawContextTrain->getData());
return arma::join_rows(DrawVisibleTrain->getData(), DrawContextTrain->getData());
}
arma::mat RbmComponent::getReconst() const
{
return to_vc(DrawVisibleReconst->getData(), DrawContextReconst->getData());
return arma::join_rows(DrawVisibleReconst->getData(), DrawContextReconst->getData());
}
void RbmComponent::trainRedraw(const arma::mat& vc)
@@ -386,7 +386,7 @@ void RbmComponent::redrawWeights(size_t index)
void RbmComponent::setTrainingData(arma::mat const& batch)
{
RbmComponent *pComp = static_cast<RbmComponent*> (root());
pComp->upDownPass(to_vc(batch, pComp->DrawContextTrain->getData()));
pComp->upDownPass(batch);
}
//[/MiscUserCode]