Skip to content
Snippets Groups Projects
Commit 073e5c2d authored by Guillaume Pasero's avatar Guillaume Pasero
Browse files

ENH: hide ML models not supporting regression

parent eb262103
No related branches found
No related tags found
No related merge requests found
......@@ -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
......
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