Made these functions work even if one the labels never shows up

in the data.

Made these functions work even if one the labels never shows up
in the data.
0a8da5ed · Davis King · 7e5cc9b7 · 0a8da5ed
Commit 0a8da5ed authored Apr 30, 2012 by Davis King
Hide whitespace changes
Inline Side-by-side

Showing with 16 additions and 4 deletions

dlib/svm/cross_validate_graph_labeling_trainer.h dlib/svm/cross_validate_graph_labeling_trainer.h +16 -4

No files found.
--- a/dlib/svm/cross_validate_graph_labeling_trainer.h
+++ b/dlib/svm/cross_validate_graph_labeling_trainer.h
@@ -59,8 +59,14 @@ namespace dlib
        }
        matrix<double, 1, 2> res;
-        res(0) = (double)num_pos_correct/(double)(num_pos); 
+        if (num_pos != 0)
-        res(1) = (double)num_neg_correct/(double)(num_neg); 
+            res(0) = (double)num_pos_correct/(double)(num_pos); 
+        else
+            res(0) = 1;
+        if (num_neg != 0)
+            res(1) = (double)num_neg_correct/(double)(num_neg); 
+        else
+            res(1) = 1;
        return res;
    }
@@ -161,8 +167,14 @@ namespace dlib
        matrix<double, 1, 2> res;
-        res(0) = (double)num_pos_correct/(double)(num_pos); 
+        if (num_pos != 0)
-        res(1) = (double)num_neg_correct/(double)(num_neg); 
+            res(0) = (double)num_pos_correct/(double)(num_pos); 
+        else
+            res(0) = 1;
+        if (num_neg != 0)
+            res(1) = (double)num_neg_correct/(double)(num_neg); 
+        else
+            res(1) = 1;
        return res;
    }