added check_input_matrices option to python bindings

src/mlpack/bindings/python/mlpack/io.pxd

@@ -38,6 +38,9 @@ cdef extern from "<mlpack/core/util/io.hpp>" namespace "mlpack" nogil:
     @staticmethod
     void ClearSettings() nogil except +
 
+    @staticmethod
+    void CheckInputMatrices() nogil except +
+
 cdef extern from "<mlpack/bindings/python/mlpack/io_util.hpp>" \
     namespace "mlpack::util" nogil:
   void SetParam[T](string, T&) nogil except +

src/mlpack/bindings/python/print_pyx.cpp

@@ -224,6 +224,16 @@ void PrintPYX(const util::BindingDetails& doc,
     cout << "  IO.SetPassed(<const string> '" << d.name << "')" << endl;
   }
 
+  // Checking the type of check_input_matrices parameter.
+  cout << "  if not isinstance(check_input_matrices, bool):" << endl;
+  cout << "    raise TypeError(" <<"\"'check_input_matrices\' must have type "
+      << "\'bool'!\")" << endl;
+  cout << endl;
+
+  // Before calling mlpackMain(), we check input matrices for NaN values if needed.
+  cout << "  if check_input_matrices:" << endl;
+  cout << "    IO.CheckInputMatrices()" << endl;
+
   // Call the method.
   cout << "  # Call the mlpack program." << endl;
   cout << "  mlpackMain()" << endl;

src/mlpack/bindings/python/py_option.hpp

@@ -64,8 +64,10 @@ class PyOption
     data.required = required;
     data.input = input;
     data.loaded = false;
-    // Only "verbose" and "copy_all_inputs" will be persistent.
-    if (identifier == "verbose" || identifier == "copy_all_inputs")
+    // Only "verbose", "copy_all_inputs" and "check_input_matrices"
+    //  will be persistent.
+    if (identifier == "verbose" || identifier == "copy_all_inputs" ||
+        identifier == "check_input_matrices")
       data.persistent = true;
     else
       data.persistent = false;

src/mlpack/core/util/io.cpp

@@ -267,3 +267,52 @@ void IO::ClearSettings()
   GetSingleton().aliases = persistentAliases;
   GetSingleton().functionMap = persistentFunctions;
 }
+
+void IO::CheckInputMatrices()
+{
+  typedef typename std::tuple<data::DatasetInfo, arma::mat> TupleType;
+  std::map<std::string, util::ParamData>::iterator itr;
+
+  for (itr = IO::Parameters().begin(); itr != IO::Parameters().end(); ++itr)
+  {
+    std::string paramName = itr->first;
+    std::string paramType = itr->second.cppType;
+    std::string errMsg = "The input " + paramName + " has NaN values.";
+    if (paramType == "arma::mat")
+    {
+      if (IO::GetParam<arma::Mat<double>>(paramName).has_nan())
+        Log::Fatal << errMsg << std::endl;
+    }
+    else if (paramType == "arma::Mat<size_t>")
+    {
+      if (IO::GetParam<arma::Mat<size_t>>(paramName).has_nan())
+        Log::Fatal << errMsg << std::endl;
+    }
+    else if (paramType == "arma::colvec")
+    {
+      if (IO::GetParam<arma::Col<double>>(paramName).has_nan())
+        Log::Fatal << errMsg << std::endl;
+    }
+    else if (paramType == "arma::Col<size_t>")
+    {
+      if (IO::GetParam<arma::Col<size_t>>(paramName).has_nan())
+        Log::Fatal << errMsg << std::endl;
+    }
+    else if (paramType == "arma::rowvec")
+    {
+      if (IO::GetParam<arma::Row<double>>(paramName).has_nan())
+        Log::Fatal << errMsg << std::endl;
+    }
+    else if (paramType == "arma::Row<size_t>")
+    {
+      if (IO::GetParam<arma::Row<size_t>>(paramName).has_nan())
+        Log::Fatal << errMsg << std::endl;
+    }
+    else if (paramType == "std::tuple<data::DatasetInfo, arma::mat>")
+    {
+      if (std::get<1>(IO::GetParam<TupleType>(paramName)).has_nan())
+        Log::Fatal << errMsg << std::endl;
+    }
+  }
+}
+

src/mlpack/core/util/io.hpp

@@ -285,6 +285,11 @@ class IO
    */
   static void ClearSettings();
 
+  /**
+   * Checks all input matrices for NaN values, if found throws an exception.
+   */
+  static void CheckInputMatrices();
+
  private:
   //! Convenience map from alias values to names.
   std::map<char, std::string> aliases;

src/mlpack/core/util/mlpack_main.hpp

@@ -230,6 +230,8 @@ PARAM_FLAG("copy_all_inputs", "If specified, all input parameters will be deep"
     " copied before the method is run.  This is useful for debugging problems "
     "where the input parameters are being modified by the algorithm, but can "
     "slow down the code.", "");
+PARAM_FLAG("check_input_matrices", "If specified, the input matrix is checked for"
+    " NaN values; an exception is thrown if any are found.", "");
 
 // Nothing else needs to be defined---the binding will use mlpackMain() as-is.