X-Git-Url: https://fleuret.org/cgi-bin/gitweb/gitweb.cgi?a=blobdiff_plain;f=mtp_example.cc;h=5f9eef247431184e9850b6a81612192506340270;hb=ddbc8b15ea4ff788e10cffd2d3977b65462dd1a7;hp=d603b3a1928c82765abaa0b6ee1c3b1ae26d3ab0;hpb=de64ce0b1101045816e6731e03d2735356c33ef0;p=mtp.git diff --git a/mtp_example.cc b/mtp_example.cc index d603b3a..5f9eef2 100644 --- a/mtp_example.cc +++ b/mtp_example.cc @@ -33,11 +33,12 @@ using namespace std; ////////////////////////////////////////////////////////////////////// -scalar_t detection_score(scalar_t a, scalar_t b, scalar_t score_noise, scalar_t flip_noise) { - if(drand48() > flip_noise) { - return a + score_noise * (2.0 * drand48() - 1.0); +scalar_t noisy_score(scalar_t true_score, scalar_t erroneous_score, + scalar_t score_noise, scalar_t flip_noise) { + if(drand48() < flip_noise) { + return erroneous_score + score_noise * (2.0 * drand48() - 1.0); } else { - return b + score_noise * (2.0 * drand48() - 1.0); + return true_score + score_noise * (2.0 * drand48() - 1.0); } } @@ -81,7 +82,7 @@ int main(int argc, char **argv) { for(int t = 0; t < nb_time_steps; t++) { for(int l = 0; l < nb_locations; l++) { - tracker->detection_scores[t][l] = detection_score(-1.0, 1.0, score_noise, flip_noise); + tracker->detection_scores[t][l] = noisy_score(-1.0, 1.0, score_noise, flip_noise); } } @@ -101,13 +102,13 @@ int main(int argc, char **argv) { if(t < nb_time_steps/2) { la = t; lb = nb_locations - 1 - t; - sa = detection_score(10.0, -1.0, score_noise, flip_noise); - sb = detection_score( 1.0, -1.0, score_noise, flip_noise); + sa = noisy_score(10.0, -1.0, score_noise, flip_noise); + sb = noisy_score( 1.0, -1.0, score_noise, flip_noise); } else { la = nb_time_steps - 1 - t; lb = t - nb_time_steps + nb_locations; - sa = detection_score( 1.0, -1.0, score_noise, flip_noise); - sb = detection_score(10.0, -1.0, score_noise, flip_noise); + sa = noisy_score( 1.0, -1.0, score_noise, flip_noise); + sb = noisy_score(10.0, -1.0, score_noise, flip_noise); } if(la > nb_locations/2 - 1) la = nb_locations/2 - 1;