don't booleanize regression values

Josh Klontz
1 parent 80f426dc
Showing 1 changed file with 2 additions and 1 deletions
openbr/plugins/svm.cpp
@@ -161,7 +161,8 @@ private:
             dst.m().at<float>(0, 0) = prediction;
             // positive values ==> first class
             // negative values ==> second class
-            prediction = prediction > 0 ? 0 : 1;
+            if (type != EPS_SVR && type != NU_SVR)
+                prediction = prediction > 0 ? 0 : 1;
         }
         if (type == EPS_SVR || type == NU_SVR)
             dst.file.set(outputVariable, prediction);