From ab96f0e380e805c4bbef9f9aec5e8a7c674cd312 Mon Sep 17 00:00:00 2001 From: Kirill Mishchenko Date: Thu, 27 Jul 2017 14:59:33 +0500 Subject: [PATCH] Treat validationSize as double --- src/mlpack/core/cv/simple_cv_impl.hpp | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/mlpack/core/cv/simple_cv_impl.hpp b/src/mlpack/core/cv/simple_cv_impl.hpp index 10a74f1896..1e72f1e7ad 100644 --- a/src/mlpack/core/cv/simple_cv_impl.hpp +++ b/src/mlpack/core/cv/simple_cv_impl.hpp @@ -149,14 +149,14 @@ size_t SimpleCV::CalculateAndAssertNumberOfTrainingPoints( const double validationSize) { - if (validationSize < 0.0F || validationSize > 1.0F) + if (validationSize < 0.0 || validationSize > 1.0) throw std::invalid_argument("SimpleCV: the validationSize parameter should " "be more than 0 and less than 1"); if (xs.n_cols < 2) throw std::invalid_argument("SimpleCV: 2 or more data points are expected"); - size_t trainingPoints = roundf(xs.n_cols * (1.0F - validationSize)); + size_t trainingPoints = round(xs.n_cols * (1.0 - validationSize)); if (trainingPoints == 0 || trainingPoints == xs.n_cols) throw std::invalid_argument("SimpleCV: the validationSize parameter is "