diff --git a/src/mlpack/methods/neighbor_search/ns_model_impl.hpp b/src/mlpack/methods/neighbor_search/ns_model_impl.hpp index 0a705626a6..075306bece 100644 --- a/src/mlpack/methods/neighbor_search/ns_model_impl.hpp +++ b/src/mlpack/methods/neighbor_search/ns_model_impl.hpp @@ -414,16 +414,16 @@ void NSModel::Search(arma::mat&& querySet, if (randomBasis) querySet = q * querySet; - Log::Info << "Searching for " << k; - if (Epsilon() != 0) - Log::Info << " approximate (e=" << Epsilon() << ")"; - Log::Info << " neighbors with "; + Log::Info << "Searching for " << k << " neighbors with "; if (!Naive() && !SingleMode()) Log::Info << "dual-tree " << TreeName() << " search..." << std::endl; else if (!Naive()) Log::Info << "single-tree " << TreeName() << " search..." << std::endl; else Log::Info << "brute-force (naive) search..." << std::endl; + if (Epsilon() != 0 && !Naive()) + Log::Info << "Maximum of " << Epsilon() * 100 << "% relative error." + << std::endl; BiSearchVisitor search(querySet, k, neighbors, distances, leafSize); @@ -436,16 +436,16 @@ void NSModel::Search(const size_t k, arma::Mat& neighbors, arma::mat& distances) { - Log::Info << "Searching for " << k; - if (Epsilon() != 0) - Log::Info << " approximate (e=" << Epsilon() << ")"; - Log::Info << " neighbors with "; + Log::Info << "Searching for " << k << " neighbors with "; if (!Naive() && !SingleMode()) Log::Info << "dual-tree " << TreeName() << " search..." << std::endl; else if (!Naive()) Log::Info << "single-tree " << TreeName() << " search..." << std::endl; else Log::Info << "brute-force (naive) search..." << std::endl; + if (Epsilon() != 0 && !Naive()) + Log::Info << "Maximum of " << Epsilon() * 100 << "% relative error." + << std::endl; MonoSearchVisitor search(k, neighbors, distances); boost::apply_visitor(search, nSearch);