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

Merge branch 'tests'

Browse Source
This commit is contained in:
Shin
2016-04-18 16:15:35 +02:00
parent 17cbf5effe 9f1f0fe763
commit 66516ceb87
14 changed files with 629 additions and 594 deletions
Download Patch File Download Diff File Expand all files Collapse all files

130
tests/feedforward.cpp
View File

@@ -1,73 +1,87 @@
#include <NeuralNetwork/FeedForward/Network.h>
#include <cassert>
#include <iostream>
#include <gtest/gtest.h>
int main() {
std::string serialized;
{ // XOR problem
NeuralNetwork::FeedForward::Network n(2);
NeuralNetwork::ActivationFunction::Sigmoid a(-1);
NeuralNetwork::FeedForward::Layer &hidden=n.appendLayer(2,a);
NeuralNetwork::FeedForward::Layer &out = n.appendLayer(1,a);
TEST(FeedForward, XOR) {
NeuralNetwork::FeedForward::Network n(2);
NeuralNetwork::ActivationFunction::Sigmoid a(-1);
NeuralNetwork::FeedForward::Layer &hidden=n.appendLayer(2,a);
NeuralNetwork::FeedForward::Layer &out = n.appendLayer(1,a);
hidden[1].weight(n[0][0])=7;
hidden[1].weight(n[0][1])=-4.7;
hidden[1].weight(n[0][2])=-4.7;
hidden[1].weight(n[0][0])=7;
hidden[1].weight(n[0][1])=-4.7;
hidden[1].weight(n[0][2])=-4.7;
hidden[2].weight(n[0][0])=2.6;
hidden[2].weight(n[0][1])=-6.4;
hidden[2].weight(n[0][2])=-6.4;
hidden[2].weight(n[0][0])=2.6;
hidden[2].weight(n[0][1])=-6.4;
hidden[2].weight(n[0][2])=-6.4;
out[1].weight(hidden[0])=-4.5;
out[1].weight(hidden[1])=9.6;
out[1].weight(hidden[2])=-6.8;
out[1].weight(hidden[0])=-4.5;
out[1].weight(hidden[1])=9.6;
out[1].weight(hidden[2])=-6.8;
{
std::vector<float> ret =n.computeOutput({1,1});
assert(ret[0] < 0.5);
}
{
std::vector<float> ret =n.computeOutput({0,1});
assert(ret[0] > 0.5);
}
{
std::vector<float> ret =n.computeOutput({1,0});
assert(ret[0] > 0.5);
}
{
std::vector<float> ret =n.computeOutput({0,0});
assert(ret[0] < 0.5);
}
serialized = n.serialize().serialize();
}
{
NeuralNetwork::FeedForward::Network *deserialized=NeuralNetwork::FeedForward::Network::Factory::deserialize(serialized).release();
std::vector<float> ret =n.computeOutput({1,1});
ASSERT_LT(ret[0], 0.5);
}
{
std::vector<float> ret =deserialized->computeOutput({1,1});
assert(ret[0] < 0.5);
}
{
std::vector<float> ret =n.computeOutput({0,1});
ASSERT_GT(ret[0], 0.5);
}
{
std::vector<float> ret =deserialized->computeOutput({0,1});
assert(ret[0] > 0.5);
}
{
std::vector<float> ret =n.computeOutput({1,0});
ASSERT_GT(ret[0], 0.5);
}
{
std::vector<float> ret =deserialized->computeOutput({1,0});
assert(ret[0] > 0.5);
}
{
std::vector<float> ret =deserialized->computeOutput({0,0});
assert(ret[0] < 0.5);
}
delete deserialized;
{
std::vector<float> ret =n.computeOutput({0,0});
ASSERT_LT(ret[0], 0.5);
}
}
TEST(FeedForward, Serialization) {
NeuralNetwork::FeedForward::Network n(2);
NeuralNetwork::ActivationFunction::Sigmoid a(-1);
NeuralNetwork::FeedForward::Layer &hidden=n.appendLayer(2,a);
NeuralNetwork::FeedForward::Layer &out = n.appendLayer(1,a);
hidden[1].weight(n[0][0])=7;
hidden[1].weight(n[0][1])=-4.7;
hidden[1].weight(n[0][2])=-4.7;
hidden[2].weight(n[0][0])=2.6;
hidden[2].weight(n[0][1])=-6.4;
hidden[2].weight(n[0][2])=-6.4;
out[1].weight(hidden[0])=-4.5;
out[1].weight(hidden[1])=9.6;
out[1].weight(hidden[2])=-6.8;
std::string serialized = n.serialize().serialize();
NeuralNetwork::FeedForward::Network *deserialized=NeuralNetwork::FeedForward::Network::Factory::deserialize(serialized).release();
{
std::vector<float> ret =deserialized->computeOutput({1,1});
ASSERT_LT(ret[0], 0.5);
}
{
std::vector<float> ret =deserialized->computeOutput({0,1});
ASSERT_GT(ret[0], 0.5);
}
{
std::vector<float> ret =deserialized->computeOutput({1,0});
ASSERT_GT(ret[0], 0.5);
}
{
std::vector<float> ret =deserialized->computeOutput({0,0});
ASSERT_LT(ret[0], 0.5);
}
delete deserialized;
}