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

corrected tests

Browse Source
This commit is contained in:
shubham1206agra
2022-05-16 13:42:10 +05:30
parent 57f53bfad3
commit ac9e7d2d4e
1 changed files with 13 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/mlpack/tests/ann_layer_test.cpp
+13 -6
View File
@@ -4903,15 +4903,22 @@ TEST_CASE("MeanPoolingTestCase", "[ANNLayerTest]")
CheckMatrices(output1, result1, 1e-1);
CheckMatrices(output2, result2, 1e-1);
output1.reshape(8, 1);
output2.reshape(6, 1);
arma::mat prev_delta1, prev_delta2;
prev_delta1 << 3.6000 << -0.9000 << arma::endr
<< 3.6000 << -0.9000 << arma::endr
<< 3.6000 << -0.9000 << arma::endr
<< 3.6000 << -0.9000 << arma::endr;
prev_delta2 << 3.6000 << -0.9000 << arma::endr
<< 3.6000 << -0.9000 << arma::endr
<< 3.6000 << -0.9000 << arma::endr;
arma::mat delta1, delta2;
delta1.set_size(8, 1);
delta2.set_size(6, 1);
module1.Backward(input, output1, delta1);
REQUIRE(arma::accu(delta1) == 25.5);
module2.Backward(input, output2, delta2);
REQUIRE(arma::accu(delta2) == 19.5);
module1.Backward(input, prev_delta1, delta1);
REQUIRE(arma::accu(delta1) == 10.8);
module2.Backward(input, prev_delta2, delta2);
REQUIRE(arma::accu(delta2) == 8.1);
}
/**