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

Treat validationSize as double

Browse Source
This commit is contained in:
Kirill Mishchenko
2017-07-27 14:59:33 +05:00
parent 5e8a4c7e79
commit ab96f0e380
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/mlpack/core/cv/simple_cv_impl.hpp
+2 -2
View File
@@ -149,14 +149,14 @@ size_t SimpleCV<MLAlgorithm,
WeightsType>::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 "