Commit 0a8da5ed authored by Davis King's avatar Davis King

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

in the data.
parent 7e5cc9b7
...@@ -59,8 +59,14 @@ namespace dlib ...@@ -59,8 +59,14 @@ namespace dlib
} }
matrix<double, 1, 2> res; matrix<double, 1, 2> res;
if (num_pos != 0)
res(0) = (double)num_pos_correct/(double)(num_pos); 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); res(1) = (double)num_neg_correct/(double)(num_neg);
else
res(1) = 1;
return res; return res;
} }
...@@ -161,8 +167,14 @@ namespace dlib ...@@ -161,8 +167,14 @@ namespace dlib
matrix<double, 1, 2> res; matrix<double, 1, 2> res;
if (num_pos != 0)
res(0) = (double)num_pos_correct/(double)(num_pos); 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); res(1) = (double)num_neg_correct/(double)(num_neg);
else
res(1) = 1;
return res; return res;
} }
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment