Skip to content

Commit

Permalink
Fix style.
Browse files Browse the repository at this point in the history
  • Loading branch information
rcurtin committed Jan 21, 2021
1 parent e2a0ac4 commit 2f504f5
Showing 1 changed file with 6 additions and 1 deletion.
7 changes: 6 additions & 1 deletion src/mlpack/methods/cf/cf_main.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -371,10 +371,15 @@ static void mlpackMain()
arma::Mat<size_t> users =
std::move(IO::GetParam<arma::Mat<size_t>>("query"));
if (users.n_rows > 1)
{
users = users.t();
}

if (users.n_rows > 1)
{
Log::Fatal << "List of query users must be one-dimensional!"
<< std::endl;
<< std::endl;
}

Log::Info << "Generating recommendations for " << users.n_elem
<< " users." << endl;
Expand Down

0 comments on commit 2f504f5

Please sign in to comment.