Commit 073e5c2d authored by Guillaume Pasero's avatar Guillaume Pasero

ENH: hide ML models not supporting regression

parent eb262103
......@@ -47,11 +47,17 @@ LearningApplicationBase<TInputValue,TOutputValue>
#ifdef OTB_USE_OPENCV
InitSVMParams();
InitBoostParams();
if (!m_RegressionFlag)
{
InitBoostParams(); // Regression not supported
}
InitDecisionTreeParams();
InitGradientBoostedTreeParams();
InitNeuralNetworkParams();
InitNormalBayesParams();
if (!m_RegressionFlag)
{
InitNormalBayesParams(); // Regression not supported
}
InitRandomForestsParams();
InitKNNParams();
#endif
......
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