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

- better error report

Browse Source 
git-svn-id: http://moon:8086/svn/software/trunk/projects/RBM@597 b431acfa-c32f-4a4a-93f1-934dc6c82436
This commit is contained in:
jens
2019-10-28 23:09:33 +00:00
parent 1d02f08242
commit 2ac8d27e75
1 changed files with 5 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
Source/Rbm.cpp
+5 -2
View File
@@ -261,11 +261,14 @@ void Rbm::train(size_t numEpochs, size_t miniBatchSize, bool &doStop)
updateHiddenBatch(); updateHiddenBatch();
MatrixXd vis = m_h * m_w.weights().transpose(); MatrixXd h = __batch * m_w.weights();
h += m_w.hiddenBias().replicate(__batch.rows(), 1);
MatrixXd vis = h * m_w.weights().transpose();
vis += m_w.visibleBias().replicate(__batch.rows(), 1); vis += m_w.visibleBias().replicate(__batch.rows(), 1);
MatrixXd diffErr = __batch - probsLogistic(vis); MatrixXd diffErr = __batch - probsLogistic(vis);
diffErr.array() *= diffErr.array(); diffErr.array() *= diffErr.array();
double err = diffErr.colwise().sum().mean(); double err = diffErr.array().mean();
cout << "error (total) = " << err << endl; cout << "error (total) = " << err << endl;
onProgressChanged(); onProgressChanged();