[mlpack-git] master: Add sanity check for #428. (db7f910)
gitdub at big.cc.gt.atl.ga.us
gitdub at big.cc.gt.atl.ga.us
Wed May 27 21:34:10 EDT 2015
Repository : https://github.com/mlpack/mlpack
On branch : master
Link : https://github.com/mlpack/mlpack/compare/3d239640213ec47c6756803bdb50df8e16b94e75...db7f910467dec872a93cf47427b3d6196fd52b39
>---------------------------------------------------------------
commit db7f910467dec872a93cf47427b3d6196fd52b39
Author: ryan <ryan at ratml.org>
Date: Wed May 27 21:16:51 2015 -0400
Add sanity check for #428.
>---------------------------------------------------------------
db7f910467dec872a93cf47427b3d6196fd52b39
.../methods/logistic_regression/logistic_regression_function.cpp | 7 +++++++
1 file changed, 7 insertions(+)
diff --git a/src/mlpack/methods/logistic_regression/logistic_regression_function.cpp b/src/mlpack/methods/logistic_regression/logistic_regression_function.cpp
index c3e1c58..f815fe0 100644
--- a/src/mlpack/methods/logistic_regression/logistic_regression_function.cpp
+++ b/src/mlpack/methods/logistic_regression/logistic_regression_function.cpp
@@ -18,6 +18,13 @@ LogisticRegressionFunction::LogisticRegressionFunction(
lambda(lambda)
{
initialPoint = arma::zeros<arma::mat>(predictors.n_rows + 1, 1);
+
+ // Sanity check.
+ if (responses.n_elem != predictors.n_cols)
+ Log::Fatal << "LogisticRegressionFunction::LogisticRegressionFunction(): "
+ << "predictors matrix has " << predictors.n_cols << " points, but "
+ << "responses vector has " << responses.n_elem << " elements (should be"
+ << " " << predictors.n_cols << ")!" << std::endl;
}
LogisticRegressionFunction::LogisticRegressionFunction(
More information about the mlpack-git
mailing list