- fixed total error calc

2024-01-31 12:52:29 +01:00
parent 2a2be76b2c
commit ccc3e0ac87
1 changed files with 1 additions and 1 deletions
@@ -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)
 	{