From c2e5ad85d1f195cfd0fb1c74cbcf9216d64b8058 Mon Sep 17 00:00:00 2001
From: David Vojtek <david.vojtek1@vsb.cz>
Date: Fri, 2 Nov 2018 12:11:11 +0100
Subject: [PATCH] Fix: fix of Gradiant descent
---
src/ErrorFunction/ErrorFunctions.cpp | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/src/ErrorFunction/ErrorFunctions.cpp b/src/ErrorFunction/ErrorFunctions.cpp
index b0e12bfd..e782851b 100644
--- a/src/ErrorFunction/ErrorFunctions.cpp
+++ b/src/ErrorFunction/ErrorFunctions.cpp
@@ -56,10 +56,10 @@ namespace lib4neuro {
void MSE::calculate_error_gradient(std::vector<double> ¶ms, std::vector<double> &grad, double alpha, size_t batch) {
size_t dim_out = this->ds->get_output_dim();
- size_t n_elements = this->ds->get_n_elements();
+
std::vector<std::pair<std::vector<double>, std::vector<double>>> data = this->ds->get_random_data_batch(batch);
-
+ size_t n_elements = data.size();
std::vector<double> error_derivative(dim_out);
--
GitLab