Update MLP.cpp

This commit is contained in:
David Nogueira
2018-09-20 13:02:56 +01:00
committed by GitHub
parent 3a3173a241
commit 75116e157e

View File

@@ -10,10 +10,7 @@
#include <fstream> #include <fstream>
#include <vector> #include <vector>
#include <algorithm> #include <algorithm>
#ifdef MLP_EASYLOGGING
#include "easylogging++.h" #include "easylogging++.h"
#endif
//desired call sintax : MLP({64*64,20,4}, {"sigmoid", "linear"}, //desired call sintax : MLP({64*64,20,4}, {"sigmoid", "linear"},
@@ -208,9 +205,9 @@ void MLP::Train(const std::vector<TrainingSample> &training_sample_set_with_bias
temp_training << predicted_output[i]; temp_training << predicted_output[i];
} }
temp_training << "]"; temp_training << "]";
#ifdef MLP_EASYLOGGING
LOG(INFO) << temp_training.str(); LOG(INFO) << temp_training.str();
#endif
} }
for (int j = 0; j < predicted_output.size(); j++) { for (int j = 0; j < predicted_output.size(); j++) {
@@ -225,15 +222,14 @@ void MLP::Train(const std::vector<TrainingSample> &training_sample_set_with_bias
learning_rate); learning_rate);
} }
#ifdef MLP_EASYLOGGING
if (output_log && ((i % (max_iterations / 10)) == 0)) if (output_log && ((i % (max_iterations / 10)) == 0))
LOG(INFO) << "Iteration " << i << " cost function f(error): " LOG(INFO) << "Iteration " << i << " cost function f(error): "
<< current_iteration_cost_function; << current_iteration_cost_function;
#endif
if (current_iteration_cost_function < min_error_cost) if (current_iteration_cost_function < min_error_cost)
break; break;
} }
#ifdef MLP_EASYLOGGING
LOG(INFO) << "Iteration " << i << " cost function f(error): " LOG(INFO) << "Iteration " << i << " cost function f(error): "
<< current_iteration_cost_function; << current_iteration_cost_function;
@@ -241,7 +237,7 @@ void MLP::Train(const std::vector<TrainingSample> &training_sample_set_with_bias
LOG(INFO) << "******* TRAINING ENDED *******"; LOG(INFO) << "******* TRAINING ENDED *******";
LOG(INFO) << "******* " << i << " iters *******"; LOG(INFO) << "******* " << i << " iters *******";
LOG(INFO) << "******************************"; LOG(INFO) << "******************************";
#endif
//{ //{
// int layer_i = -1; // int layer_i = -1;
// int node_i = -1; // int node_i = -1;