Skip to content
Snippets Groups Projects
Commit 3a9b6546 authored by Martin Beseda's avatar Martin Beseda
Browse files

FIX: Modified to use smart pointer to DataSet

parent 704182d6
No related branches found
No related tags found
No related merge requests found
......@@ -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
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment