Mirrors/MLP
1
0
Fork 0
mirror of https://github.com/davidalbertonogueira/MLP.git synced 2025-12-18 21:04:40 +03:00
Code Issues Actions Packages Projects Releases Wiki Activity

Edit README and Makefile.

Browse Source
This commit is contained in:
davidjacnogueira
2017-03-11 14:20:49 +00:00
parent 4005aff367
commit d44ec01e86
9 changed files with 101 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
src/IrisDatasetTest.cpp
View File

@@ -115,19 +115,16 @@ int main(int argc, char *argv[]) {
}
{
/* 4 inputs + 1 bias.
* 1 hidden layer(s) of 4 neurons.
* 3 outputs (1 per iris_class)
*/
// 4 inputs + 1 bias.
// 1 hidden layer(s) of 4 neurons.
// 3 outputs (1 per iris_class)
MLP my_mlp({ 4 + 1, 4 ,3 }, { "sigmoid", "linear" }, false);
int loops = 5000;
// Train the network with backpropagation.
LOG(INFO) << "Training for " << loops << " loops over data.";
my_mlp.UpdateMiniBatch(training_sample_set_with_bias, .01, loops, 0.10, false);
my_mlp.Train(training_sample_set_with_bias, .01, loops, 0.10, false);
my_mlp.SaveMLPNetwork(std::string("../../data/iris.mlp"));
}