From 3a9b6546fa18f70282cdda38e9a48d16ca6845ef Mon Sep 17 00:00:00 2001
From: Martin Beseda <martin.beseda@vsb.cz>
Date: Wed, 20 Mar 2019 03:33:39 +0100
Subject: [PATCH] FIX: Modified to use smart pointer to DataSet
---
src/examples/x2_fitting.cpp | 6 +++---
1 file changed, 3 insertions(+), 3 deletions(-)
diff --git a/src/examples/x2_fitting.cpp b/src/examples/x2_fitting.cpp
index 30642211..2d9bb56d 100644
--- a/src/examples/x2_fitting.cpp
+++ b/src/examples/x2_fitting.cpp
@@ -13,20 +13,20 @@ int main() {
std::vector<unsigned int> input_ind = {0};
std::vector<unsigned int> output_ind = {1};
- l4n::DataSet ds = reader.get_data_set(&input_ind, &output_ind);
+ std::shared_ptr<l4n::DataSet> ds = reader.get_data_set(&input_ind, &output_ind);
std::vector<unsigned int> neuron_numbers_in_layers = {1, 15, 1};
std::vector<l4n::NEURON_TYPE> hidden_type_v = {l4n::NEURON_TYPE::LOGISTIC};
l4n::FullyConnectedFFN net(&neuron_numbers_in_layers, &hidden_type_v);
- l4n::MSE mse(&net, &ds);
+ l4n::MSE mse(&net, ds.get());
l4n::GradientDescent gs(1e-5, 20, 200000);
net.randomize_parameters();
gs.optimize(mse);
- mse.eval_on_data_set(&ds);
+ mse.eval_on_data_set(ds.get());
return 0;
}
\ No newline at end of file
--
GitLab