shin/NeuralNetworkLib
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

added new tests

Browse Source
This commit is contained in:
Tomáš Černík
2016-02-07 22:58:05 +01:00
parent fcc663f0ba
commit 782ac8d1e8
1 changed files with 72 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

72
tests/backpropagation.cpp
View File

@@ -41,4 +41,76 @@ int main() {
assert(ret[0] < 0.1); assert(ret[0] < 0.1);
} }
} }
{ // AND problem
NeuralNetwork::FeedForward::Network n(2);
NeuralNetwork::ActivationFunction::Sigmoid a(-1);
n.appendLayer(2,a);
n.appendLayer(1,a);
n.randomizeWeights();
NeuralNetwork::Learning::BackPropagation prop;
for(int i=0;i<10000;i++) {
prop.teach(n,{1,1},{1});
prop.teach(n,{0,0},{0});
prop.teach(n,{0,1},{0});
prop.teach(n,{1,0},{0});
}
{
std::vector<float> ret =n.computeOutput({1,1});
assert(ret[0] > 0.9);
}
{
std::vector<float> ret =n.computeOutput({0,1});
assert(ret[0] < 0.1);
}
{
std::vector<float> ret =n.computeOutput({1,0});
assert(ret[0] < 0.1);
}
{
std::vector<float> ret =n.computeOutput({0,0});
assert(ret[0] < 0.1);
}
}
{ // NOT AND problem
NeuralNetwork::FeedForward::Network n(2);
NeuralNetwork::ActivationFunction::Sigmoid a(-1);
n.appendLayer(2,a);
n.appendLayer(1,a);
n.randomizeWeights();
NeuralNetwork::Learning::BackPropagation prop;
for(int i=0;i<10000;i++) {
prop.teach(n,{1,1},{0});
prop.teach(n,{0,0},{1});
prop.teach(n,{0,1},{1});
prop.teach(n,{1,0},{1});
}
{
std::vector<float> ret =n.computeOutput({1,1});
assert(ret[0] < 0.1);
}
{
std::vector<float> ret =n.computeOutput({0,1});
assert(ret[0] > 0.9);
}
{
std::vector<float> ret =n.computeOutput({1,0});
assert(ret[0] > 0.9);
}
{
std::vector<float> ret =n.computeOutput({0,0});
assert(ret[0] > 0.9);
}
}
} }