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

Merge branch 'sanity_checks' of https://github.com/NippunSharma/mlpack into sanity_checks

Browse Source
This commit is contained in:
NippunSharma
2021-01-24 00:37:10 +05:30
parent 723ecf82eb 33af71dbce
commit 0c60d8c423
1 changed files with 21 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/mlpack/core/util/io.cpp
+21 -11
View File
@@ -279,26 +279,36 @@ void IO::CheckInputMatrices()
std::string paramType = itr->second.cppType;
if (paramType == "arma::mat")
IO::CheckInputMatrix<arma::Mat<double>>(
IO::GetParam<arma::Mat<double>>(paramName), paramName);
{
IO::CheckInputMatrix(IO::GetParam<arma::mat>(paramName), paramName);
}
else if (paramType == "arma::Mat<size_t>")
IO::CheckInputMatrix<arma::Mat<size_t>>(
{
IO::CheckInputMatrix(
IO::GetParam<arma::Mat<size_t>>(paramName), paramName);
}
else if (paramType == "arma::colvec")
IO::CheckInputMatrix<arma::Col<double>>(
IO::GetParam<arma::Col<double>>(paramName), paramName);
{
IO::CheckInputMatrix(IO::GetParam<arma::colvec>(paramName), paramName);
}
else if (paramType == "arma::Col<size_t>")
IO::CheckInputMatrix<arma::Col<size_t>>(
{
IO::CheckInputMatrix(
IO::GetParam<arma::Col<size_t>>(paramName), paramName);
}
else if (paramType == "arma::rowvec")
IO::CheckInputMatrix<arma::Row<double>>(
IO::GetParam<arma::Row<double>>(paramName), paramName);
{
IO::CheckInputMatrix(IO::GetParam<arma::rowvec>(paramName), paramName);
}
else if (paramType == "arma::Row<size_t>")
IO::CheckInputMatrix<arma::Row<size_t>>(
{
IO::CheckInputMatrix(
IO::GetParam<arma::Row<size_t>>(paramName), paramName);
}
else if (paramType == "std::tuple<data::DatasetInfo, arma::mat>")
IO::CheckInputMatrix<arma::Mat<double>>(
{
IO::CheckInputMatrix(
std::get<1>(IO::GetParam<TupleType>(paramName)), paramName);
}
}
}