diff --git a/src/mlpack/methods/softmax_regression/softmax_regression.hpp b/src/mlpack/methods/softmax_regression/softmax_regression.hpp index a547ac61e4..25f96e8ed3 100644 --- a/src/mlpack/methods/softmax_regression/softmax_regression.hpp +++ b/src/mlpack/methods/softmax_regression/softmax_regression.hpp @@ -111,7 +111,7 @@ class SoftmaxRegression * @param testData Matrix of data points for which predictions are to be made. * @param predictions Vector to store the predictions in. */ - void Predict(const arma::mat& testData, arma::vec& predictions); + void Predict(const arma::mat& testData, arma::vec& predictions) const; /** * Computes accuracy of the learned model given the feature data and the diff --git a/src/mlpack/methods/softmax_regression/softmax_regression_impl.hpp b/src/mlpack/methods/softmax_regression/softmax_regression_impl.hpp index 43f6e016e3..db4d890bfb 100644 --- a/src/mlpack/methods/softmax_regression/softmax_regression_impl.hpp +++ b/src/mlpack/methods/softmax_regression/softmax_regression_impl.hpp @@ -58,7 +58,7 @@ SoftmaxRegression::SoftmaxRegression( template class OptimizerType> void SoftmaxRegression::Predict(const arma::mat& testData, - arma::vec& predictions) + arma::vec& predictions) const { // Calculate the probabilities for each test input. arma::mat hypothesis, probabilities;