Skip to content
Snippets Groups Projects
Commit 29175036 authored by Julien Malik's avatar Julien Malik
Browse files

ENH: ListSampleGenerator, use the precomputed m_ClassMinSize

parent 281dd33e
No related branches found
No related tags found
No related merge requests found
......@@ -299,17 +299,7 @@ ListSampleGenerator<TImage, TVectorData>
m_ClassesProbValidation.clear();
//Go through the classes size to find the smallest one
double minSizeTraining = -1;
for (std::map<ClassLabelType, double>::const_iterator itmap = m_ClassesSize.begin();
itmap != m_ClassesSize.end();
++itmap)
{
if ((minSizeTraining < 0) || (minSizeTraining > itmap->second))
{
minSizeTraining = itmap->second;
}
}
double minSizeTraining = m_ClassMinSize;
double minSizeValidation = minSizeTraining;
//Apply the proportion between training and validation samples (all training by default)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment