OrgLion-ML
/
dlib
mirror of https://github.com/davisking/dlib.git
1
0
Fork 0
Code Issues Releases Wiki Activity

Made the loss calculation more explicit.

This commit is contained in:
Davis King 2014-04-05 18:56:50 -04:00
parent 137b7f80c8
commit 67448f72f7
1 changed files with 10 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
dlib/svm/structural_svm_assignment_problem.h
View File

@ -193,6 +193,8 @@ namespace dlib
} }
cost.set_size(size, size); cost.set_size(size, size);
const double loss_for_false_association = 1;
const double loss_for_missed_association = 1;
typename feature_extractor::feature_vector_type feats; typename feature_extractor::feature_vector_type feats;
// now fill out the cost assignment matrix // now fill out the cost assignment matrix
@ -211,7 +213,7 @@ namespace dlib
// add in the loss since this corresponds to an incorrect prediction. // add in the loss since this corresponds to an incorrect prediction.
if (c != labels[idx][r]) if (c != labels[idx][r])
{ {
cost(r,c) += 1; cost(r,c) += loss_for_false_association;
} }
} }
else else
@ -219,7 +221,7 @@ namespace dlib
if (labels[idx][r] == -1) if (labels[idx][r] == -1)
cost(r,c) = 0; cost(r,c) = 0;
else else
cost(r,c) = 1; // 1 for the loss cost(r,c) = loss_for_missed_association;
} }
} }
@ -250,7 +252,12 @@ namespace dlib
assignment[i] = -1; assignment[i] = -1;
if (assignment[i] != labels[idx][i]) if (assignment[i] != labels[idx][i])
loss += 1; {
if (assignment[i] == -1)
loss += loss_for_missed_association;
else
loss += loss_for_false_association;
}
} }
get_joint_feature_vector(samples[idx], assignment, psi); get_joint_feature_vector(samples[idx], assignment, psi);