apaam/mlpack
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fixed Code style issues

Browse Source
This commit is contained in:
Vaibhav Pathak
2023-04-02 20:36:14 +05:30
parent c6444e93d1
commit 21e9d32d06
1 changed files with 30 additions and 27 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/mlpack/tests/ann/layer/hard_tanh.cpp
+30 -27
View File
@@ -22,34 +22,37 @@ using namespace mlpack;
TEST_CASE("SimpleHardTanHTest", "[ANNLayerTest]")
{
arma::mat output, gy, g;
arma::mat input = {{-1.3743, -0.5565, 0.2742, -0.0151, -1.4871},
{1.5797, -4.2711, -2.2505, -1.7105, -1.2544},
{0.4023, 0.5676, 2.3100, 1.6658, -0.1907},
{0.1897, 0.9097, 0.1418, -1.5349, 0.1225},
{-0.1101, -3.3656, -5.4033, -2.2240, -3.3235}};
arma::mat actualOutput = {{-1.0000, -0.5565, 0.2742, -0.0151, -1.0000},
{1.0000, -1.0000, -1.0000, -1.0000, -1.0000},
{0.4023, 0.5676, 1.0000, 1.0000, -0.1907},
{0.1897, 0.9097, 0.1418, -1.0000, 0.1225},
{-0.1101, -1.0000, -1.0000, -1.0000, -1.0000}};
arma::mat output, gy, g;
arma::mat input = {{-1.3743, -0.5565, 0.2742, -0.0151, -1.4871},
{1.5797, -4.2711, -2.2505, -1.7105, -1.2544},
{0.4023, 0.5676, 2.3100, 1.6658, -0.1907},
{0.1897, 0.9097, 0.1418, -1.5349, 0.1225},
{-0.1101, -3.3656, -5.4033, -2.2240, -3.3235}};
arma::mat actualOutput = {{-1.0000, -0.5565, 0.2742, -0.0151, -1.0000},
{1.0000, -1.0000, -1.0000, -1.0000, -1.0000},
{0.4023, 0.5676, 1.0000, 1.0000, -0.1907},
{0.1897, 0.9097, 0.1418, -1.0000, 0.1225},
{-0.1101, -1.0000, -1.0000, -1.0000, -1.0000}};
HardTanH module;
HardTanH module;
output.set_size(5,5);
// Test the Forward function
module.Forward(input, output);
REQUIRE(arma::accu(output - actualOutput) == Approx(0).epsilon(1e-4));
output.set_size(5,5);
// Test the Forward function
module.Forward(input, output);
REQUIRE(arma::accu(output - actualOutput) == Approx(0).epsilon(1e-4));
arma::mat delta = {{0 , 1.0, 1.0, 1.0, 0.0},
{0 , 0 , 0 , 0.0, 0.0},
{1.0, 1.0, 0 , 0.0, 1.0},
{1.0, 1.0, 1.0, 0.0, 1.0},
{1.0, 0 , 0.0, 0.0, 0.0}};
arma::mat delta = {{0 , 1.0, 1.0, 1.0, 0.0},
{0 , 0 , 0 , 0.0, 0.0},
{1.0, 1.0, 0 , 0.0, 1.0},
{1.0, 1.0, 1.0, 0.0, 1.0},
{1.0, 0 , 0.0, 0.0, 0.0}};
gy.set_size(5,5);
gy.fill(1);
g.set_size(5,5);
//Test the Backward function
module.Backward(output, gy, g);
REQUIRE(arma::accu(g - delta) == Approx(0).epsilon(1e-4));
gy.set_size(5,5);
gy.fill(1);
g.set_size(5,5);
//Test the Backward function
module.Backward(output, gy, g);
REQUIRE(arma::accu(g - delta) == Approx(0).epsilon(1e-4));
}