Commit db9711e4 authored by remi cresson's avatar remi cresson
Browse files

FIX: validation step not performed if not user parameters are set

parent d9f71201
......@@ -468,8 +468,6 @@ public:
AddProcess(m_TrainModelFilter, "Training epoch #" + std::to_string(epoch));
m_TrainModelFilter->Update();
if (do_validation)
{
// Validate the model
if (epoch % GetParameterInt("validation.step") == 0)
{
......@@ -494,6 +492,8 @@ public:
PrintClassificationMetrics(m_ValidateModelFilter->GetConfusionMatrix(i), m_ValidateModelFilter->GetMapOfClasses(i));
}
if (do_validation)
{
// 2. Evaluate the metrics against the validation data
// Here we just change the input sources and references
......@@ -514,7 +514,7 @@ public:
PrintClassificationMetrics(m_ValidateModelFilter->GetConfusionMatrix(i), m_ValidateModelFilter->GetMapOfClasses(i));
}
} // Step is OK to perform validation
}
} // Do the validation against the validation data
} // Next epoch
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment