Skip to content
Snippets Groups Projects
Commit a18cf8c6 authored by Jordi Inglada's avatar Jordi Inglada
Browse files

ENH: find KNN in parallel for smote

parent ef3973f3
No related branches found
No related tags found
Loading
...@@ -147,6 +147,7 @@ void findKNNIndices(const SampleVectorType& inSamples, ...@@ -147,6 +147,7 @@ void findKNNIndices(const SampleVectorType& inSamples,
{ {
const auto nbSamples = inSamples.size(); const auto nbSamples = inSamples.size();
nnVector.resize(nbSamples); nnVector.resize(nbSamples);
#pragma omp parallel for
for(size_t sampleIdx=0; sampleIdx<nbSamples; ++sampleIdx) for(size_t sampleIdx=0; sampleIdx<nbSamples; ++sampleIdx)
{ {
NNIndicesType nns; NNIndicesType nns;
......
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