111 lines
3.3 KiB
C++
111 lines
3.3 KiB
C++
#include <NeuralNetwork/Network.h>
|
|
#include <NeuralNetwork/ActivationFunction/Heaviside.h>
|
|
#include <NeuralNetwork/ActivationFunction/Sigmoid.h>
|
|
#include <NeuralNetwork/ActivationFunction/HyperbolicTangent.h>
|
|
#include <NeuralNetwork/ActivationFunction/Linear.h>
|
|
|
|
#pragma GCC diagnostic push
|
|
#pragma GCC diagnostic ignored "-Weffc++"
|
|
|
|
#include <gtest/gtest.h>
|
|
|
|
#pragma GCC diagnostic pop
|
|
|
|
union SSE {
|
|
__m128 sse; // SSE 4 x float vector
|
|
float floats[4]; // scalar array of 4 floats
|
|
};
|
|
|
|
NEURAL_NETWORK_INIT();
|
|
|
|
TEST(Heaviside, ParamOne) {
|
|
NeuralNetwork::ActivationFunction::Heaviside h(1.0);
|
|
ASSERT_EQ(h(0.2), 0);
|
|
ASSERT_EQ(h(1.2), 1);
|
|
}
|
|
|
|
TEST(Heaviside, ParamZeroPointSeven) {
|
|
NeuralNetwork::ActivationFunction::Heaviside h(0.7);
|
|
ASSERT_EQ(h(0.2), 0);
|
|
ASSERT_EQ(h(0.8), 1);
|
|
}
|
|
|
|
TEST(Sigmoid, ParamZeroPointSeven) {
|
|
NeuralNetwork::ActivationFunction::Sigmoid s(0.7);
|
|
ASSERT_GT(s(0.1), 0.482407);
|
|
ASSERT_LT(s(0.1), 0.482607);
|
|
|
|
ASSERT_GT(s(10), 0.000901051);
|
|
ASSERT_LT(s(10), 0.000921051);
|
|
}
|
|
|
|
TEST(Sigmoid, ParamMinusFive) {
|
|
NeuralNetwork::ActivationFunction::Sigmoid s(-5);
|
|
ASSERT_GT(s(0.1), 0.622359);
|
|
ASSERT_LT(s(0.1), 0.622559);
|
|
|
|
ASSERT_GT(s(0.7), 0.970588);
|
|
ASSERT_LT(s(0.7), 0.970788);
|
|
}
|
|
|
|
TEST(SigmoidSSE, ParamMinusZeroPointSeven) {
|
|
NeuralNetwork::ActivationFunction::Sigmoid s(-0.7);
|
|
SSE comp;
|
|
comp.floats[0] = 0.1;
|
|
comp.floats[1] = 10;
|
|
comp.sse = s(comp.sse);
|
|
|
|
ASSERT_GT(comp.floats[0], 0.517483);
|
|
ASSERT_LT(comp.floats[0], 0.51750);
|
|
|
|
ASSERT_GT(comp.floats[1], 0.998989);
|
|
ASSERT_LT(comp.floats[1], 0.999189);
|
|
}
|
|
|
|
TEST(Linear, ParamOne) {
|
|
NeuralNetwork::ActivationFunction::Linear s(1.0);
|
|
ASSERT_GT(s(0.5), 0.4999);
|
|
ASSERT_LT(s(0.5), 0.5001);
|
|
|
|
ASSERT_EQ(s(0.0), 0.0);
|
|
}
|
|
|
|
TEST(Linear, ParamZeroPointSeven) {
|
|
NeuralNetwork::ActivationFunction::Linear s(0.7);
|
|
ASSERT_GT(s(1.0), 0.6999);
|
|
ASSERT_LT(s(1.0), 0.7001);
|
|
|
|
ASSERT_EQ(s(0.0), 0.0);
|
|
}
|
|
|
|
TEST(Linear, Serialize) {
|
|
NeuralNetwork::ActivationFunction::Linear l(2.5);
|
|
const std::string tmp = l.serialize().serialize();
|
|
NeuralNetwork::ActivationFunction::ActivationFunction* deserialized = NeuralNetwork::ActivationFunction::Factory::deserialize(l.serialize()).release();
|
|
ASSERT_EQ(tmp, deserialized->serialize().serialize());
|
|
delete deserialized;
|
|
}
|
|
|
|
TEST(Heaviside, Serialize) {
|
|
NeuralNetwork::ActivationFunction::Heaviside l(2.5);
|
|
const std::string tmp = l.serialize().serialize();
|
|
NeuralNetwork::ActivationFunction::ActivationFunction* deserialized = NeuralNetwork::ActivationFunction::Factory::deserialize(l.serialize()).release();
|
|
ASSERT_EQ(tmp, deserialized->serialize().serialize());
|
|
delete deserialized;
|
|
}
|
|
|
|
TEST(HyperbolicTangent, Serialize) {
|
|
NeuralNetwork::ActivationFunction::HyperbolicTangent l(2.5);
|
|
const std::string tmp = l.serialize().serialize();
|
|
NeuralNetwork::ActivationFunction::ActivationFunction* deserialized = NeuralNetwork::ActivationFunction::Factory::deserialize(l.serialize()).release();
|
|
ASSERT_EQ(tmp, deserialized->serialize().serialize());
|
|
delete deserialized;
|
|
}
|
|
|
|
TEST(Sigmoid, Serialize) {
|
|
NeuralNetwork::ActivationFunction::Sigmoid l(2.5);
|
|
const std::string tmp = l.serialize().serialize();
|
|
NeuralNetwork::ActivationFunction::ActivationFunction* deserialized = NeuralNetwork::ActivationFunction::Factory::deserialize(l.serialize()).release();
|
|
ASSERT_EQ(tmp, deserialized->serialize().serialize());
|
|
delete deserialized;
|
|
} |