From ccc3e0ac8794f5ac3bec55a4302611bb3bfbb2a6 Mon Sep 17 00:00:00 2001 From: jens Date: Wed, 31 Jan 2024 12:52:29 +0100 Subject: [PATCH] - fixed total error calc --- source/Rbm.cpp | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/source/Rbm.cpp b/source/Rbm.cpp index 83e9bf2..24d8022 100644 --- a/source/Rbm.cpp +++ b/source/Rbm.cpp @@ -370,7 +370,7 @@ void Rbm::train(arma::mat const &batch, IListener* pListener) } // number of mini batches // Update final status - status.err_total = rms_error_accu(batch - prob(h_to_v(prob(v_to_h(batch))))); + status.err_total = rms_error_accu(batch - toVisibleProbs(toHiddenProbs(batch))); if (pListener) {