Minor change

This commit is contained in:
Prabhat
2018-03-05 03:27:54 +05:30
parent d26a3c14bd
commit 69700498a5
2 changed files with 4 additions and 3 deletions
@@ -118,7 +118,8 @@ class GlorotInitialization
/**
* XavierInitilization is the popular name for this method.
*/
using XavierInitialization = GlorotInitialization<false>
using XavierInitialization = GlorotInitialization<true>;
using GlorotInitialization = GlorotInitialization<false>;
} // namespace ann
} // namespace mlpack
+2 -2
View File
@@ -294,7 +294,7 @@ BOOST_AUTO_TEST_CASE(GlorotInitUniformTest)
arma::mat weights;
arma::cube weights3d;
GlorotInitialization<true> glorotInit;
XavierInitialization glorotInit;
glorotInit.Initialize(weights, 100, 100);
glorotInit.Initialize(weights3d, 100, 100, 2);
@@ -315,7 +315,7 @@ BOOST_AUTO_TEST_CASE(GlorotInitNormalTest)
arma::mat weights;
arma::cube weights3d;
GlorotInitialization<false> glorotInit;
GlorotInitialization glorotInit;
glorotInit.Initialize(weights, 100, 100);
glorotInit.Initialize(weights3d, 100, 100, 2);