|
@@ -4,7 +4,12 @@
|
|
|
* @author Frank Prüfer
|
|
|
* @date 08/29/2013
|
|
|
|
|
|
-*/
|
|
|
+*/
|
|
|
+
|
|
|
+#ifdef NICE_USELIB_OPENMP
|
|
|
+#include <omp.h>
|
|
|
+#endif
|
|
|
+
|
|
|
#include <iostream>
|
|
|
|
|
|
#include "vislearning/regression/npregression/RegKNN.h"
|
|
@@ -68,6 +73,7 @@ double RegKNN::predict ( const NICE::Vector & x )
|
|
|
exit(-1);
|
|
|
}
|
|
|
|
|
|
+#pragma omp parallel for
|
|
|
for(uint i = 0; i < dataSet.size(); i++){
|
|
|
|
|
|
double distance = distancefunc->calculate (x,dataSet[i]);
|
|
@@ -77,6 +83,7 @@ double RegKNN::predict ( const NICE::Vector & x )
|
|
|
cerr << x << endl;
|
|
|
cerr << dataSet[i] << endl;
|
|
|
}
|
|
|
+// #pragma omp critical
|
|
|
distances[i] = distance;
|
|
|
}
|
|
|
|