diff --git a/src/mlpack/methods/ann/layer/softmin_impl.hpp b/src/mlpack/methods/ann/layer/softmin_impl.hpp index 6082a31b2a..7693ca11dd 100644 --- a/src/mlpack/methods/ann/layer/softmin_impl.hpp +++ b/src/mlpack/methods/ann/layer/softmin_impl.hpp @@ -43,7 +43,7 @@ void Softmin::Backward( const arma::Mat& gy, arma::Mat& g) { - g = input % (arma::repmat(arma::sum(gy % input), input.n_rows, 1) - gy); + g = input % (gy - arma::repmat(arma::sum(gy % input), input.n_rows, 1)); } template diff --git a/src/mlpack/tests/main_tests/mean_shift_test.cpp b/src/mlpack/tests/main_tests/mean_shift_test.cpp index eea3ceb3c7..352622a738 100644 --- a/src/mlpack/tests/main_tests/mean_shift_test.cpp +++ b/src/mlpack/tests/main_tests/mean_shift_test.cpp @@ -19,8 +19,8 @@ static const std::string testName = "MeanShift"; #include #include "test_helper.hpp" -#include -#include "../test_tools.hpp" +#include "../test_catch_tools.hpp" +#include "catch.hpp" using namespace mlpack;