apaam/mlpack
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

added templated utility function

Browse Source
This commit is contained in:
Nippun Sharma
2021-01-13 12:05:12 +05:30
parent 7056aebcdf
commit 5e0a3472cd
3 changed files with 28 additions and 51 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/mlpack/core/util/io.cpp
+7 -51
View File
@@ -277,65 +277,21 @@ void IO::CheckInputMatrices()
{
std::string paramName = itr->first;
std::string paramType = itr->second.cppType;
std::string errMsg1 = "The input " + paramName + " has NaN values.";
std::string errMsg2 = "The input " + paramName + " has inf values.";
if (paramType == "arma::mat")
{
if (IO::GetParam<arma::Mat<double>>(paramName).has_nan())
Log::Fatal << errMsg1 << std::endl;
if (IO::GetParam<arma::Mat<double>>(paramName).has_inf())
Log::Fatal << errMsg2 << std::endl;
}
IO::CheckInputMatrix<arma::Mat<double>>(paramName);
else if (paramType == "arma::Mat<size_t>")
{
if (IO::GetParam<arma::Mat<size_t>>(paramName).has_nan())
Log::Fatal << errMsg1 << std::endl;
if (IO::GetParam<arma::Mat<size_t>>(paramName).has_inf())
Log::Fatal << errMsg2 << std::endl;
}
IO::CheckInputMatrix<arma::Mat<size_t>>(paramName);
else if (paramType == "arma::colvec")
{
if (IO::GetParam<arma::Col<double>>(paramName).has_nan())
Log::Fatal << errMsg1 << std::endl;
if (IO::GetParam<arma::Col<double>>(paramName).has_inf())
Log::Fatal << errMsg2 << std::endl;
}
IO::CheckInputMatrix<arma::Col<double>>(paramName);
else if (paramType == "arma::Col<size_t>")
{
if (IO::GetParam<arma::Col<size_t>>(paramName).has_nan())
Log::Fatal << errMsg1 << std::endl;
if (IO::GetParam<arma::Col<size_t>>(paramName).has_inf())
Log::Fatal << errMsg2 << std::endl;
}
IO::CheckInputMatrix<arma::Col<size_t>>(paramName);
else if (paramType == "arma::rowvec")
{
if (IO::GetParam<arma::Row<double>>(paramName).has_nan())
Log::Fatal << errMsg1 << std::endl;
if (IO::GetParam<arma::Row<double>>(paramName).has_inf())
Log::Fatal << errMsg2 << std::endl;
}
IO::CheckInputMatrix<arma::Row<double>>(paramName);
else if (paramType == "arma::Row<size_t>")
{
if (IO::GetParam<arma::Row<size_t>>(paramName).has_nan())
Log::Fatal << errMsg1 << std::endl;
if (IO::GetParam<arma::Row<size_t>>(paramName).has_inf())
Log::Fatal << errMsg2 << std::endl;
}
IO::CheckInputMatrix<arma::Row<size_t>>(paramName);
else if (paramType == "std::tuple<data::DatasetInfo, arma::mat>")
{
if (std::get<1>(IO::GetParam<TupleType>(paramName)).has_nan())
Log::Fatal << errMsg1 << std::endl;
if (std::get<1>(IO::GetParam<TupleType>(paramName)).has_inf())
Log::Fatal << errMsg2 << std::endl;
}
IO::CheckInputMatrix<TupleType>(paramName);
}
}
src/mlpack/core/util/io.hpp
+8
View File
@@ -285,6 +285,14 @@ class IO
*/
static void ClearSettings();
/**
* Utility function for CheckInputMatrices().
*
* @param matrix Matrix to check for NaN or Inf values.
*/
template<typename T>
static void CheckInputMatrix(T& matrix);
/**
* Checks all input matrices for NaN and inf values, exits if found any.
*/
src/mlpack/core/util/io_impl.hpp
+13
View File
@@ -145,6 +145,19 @@ T& IO::GetRawParam(const std::string& identifier)
}
}
template<typename T>
void CheckInputMatrix(const std::string paramName)
{
std::string errMsg1 = "The input " + paramName + " has NaN values.";
std::string errMsg2 = "The input " + paramName + " has inf values.";
if (IO::GetParam<T>(paramName).has_nan())
Log::Fatal << errMsg1 << std::endl;
if (IO::GetParam<T>(paramName).has_inf())
Log::Fatal << errMsg2 << std::endl;
}
} // namespace mlpack
#endif