Skip to content
Snippets Groups Projects
Commit 237a8cc1 authored by Jordi Inglada's avatar Jordi Inglada Committed by Manuel Grizonnet
Browse files

ENH: move proba computation inside condition

parent 743a9dcc
No related branches found
No related tags found
No related merge requests found
...@@ -158,10 +158,9 @@ SharkRandomForestsMachineLearningModel<TInputValue,TOutputValue> ...@@ -158,10 +158,9 @@ SharkRandomForestsMachineLearningModel<TInputValue,TOutputValue>
omp_set_num_threads(itk::MultiThreader::GetGlobalDefaultNumberOfThreads()); omp_set_num_threads(itk::MultiThreader::GetGlobalDefaultNumberOfThreads());
#endif #endif
shark::Data<shark::RealVector> probas = m_RFModel(inputSamples);
if(quality != ITK_NULLPTR) if(quality != ITK_NULLPTR)
{ {
shark::Data<shark::RealVector> probas = m_RFModel(inputSamples);
unsigned int id = startIndex; unsigned int id = startIndex;
for(shark::RealVector && p : probas.elements()) for(shark::RealVector && p : probas.elements())
{ {
......
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