diff --git a/src/mlpack/methods/ann/layer/vr_class_reward_layer.hpp b/src/mlpack/methods/ann/layer/vr_class_reward_layer.hpp index 9b1451c0b4..09e55b5acd 100644 --- a/src/mlpack/methods/ann/layer/vr_class_reward_layer.hpp +++ b/src/mlpack/methods/ann/layer/vr_class_reward_layer.hpp @@ -70,7 +70,6 @@ class VRClassRewardLayer template double Forward(const arma::Mat& input, const arma::Mat& target) { - double output = 0; reward = 0; arma::uword index; @@ -78,7 +77,7 @@ class VRClassRewardLayer { input.unsafe_col(i).max(index); reward = ((index + 1) == target(i)) * scale; - } + } if (sizeAverage) { @@ -101,7 +100,7 @@ class VRClassRewardLayer double Backward(const arma::field >& input, const arma::Mat& /* gy */, arma::field >& g) - { + { g = arma::field >(2, 1); g(0, 0) = arma::zeros(input(0, 0).n_rows, input(0, 0).n_cols);