diff --git a/error_correction/src/macaon_train_error_detector.cpp b/error_correction/src/macaon_train_error_detector.cpp index da5f3f5432eca467a1af4f941286ff683fdb33ea..1751e92eac409a517dcf7df1958ac06b2968f135 100644 --- a/error_correction/src/macaon_train_error_detector.cpp +++ b/error_correction/src/macaon_train_error_detector.cpp @@ -265,21 +265,25 @@ std::map<std::string, std::pair<float, std::pair<float, float> > > getScoreOnDev } } - fprintf(stderr, "Class 0 nbExemples : %d\n", pred0Hyp0+pred1Hyp0); + int nbErrorsIntroduced = pred1Hyp0; + int nbErrorsCorrected = pred1Hyp1; + + fprintf(stderr, "\nClass 0 nbExemples : %d\n", pred0Hyp0+pred1Hyp0); fprintf(stderr, "Class 0 precision : %.2f%%\n", 100.0*pred0Hyp0 / (pred0Hyp0+pred0Hyp1)); fprintf(stderr, "Class 0 recall : %.2f%%\n\n", 100.0*pred0Hyp0 / (pred0Hyp0+pred1Hyp0)); fprintf(stderr, "Class 1 nbExemples : %d\n", pred0Hyp1+pred1Hyp1); fprintf(stderr, "Class 1 precision : %.2f%%\n", 100.0*pred1Hyp1 / (pred1Hyp1+pred1Hyp0)); - fprintf(stderr, "Class 1 recall : %.2f%%\n", 100.0*pred1Hyp1 / (pred1Hyp1+pred0Hyp1)); + fprintf(stderr, "Class 1 recall : %.2f%%\n\n", 100.0*pred1Hyp1 / (pred1Hyp1+pred0Hyp1)); + + fprintf(stderr, "Nb errors introduced : %d\n", nbErrorsIntroduced); + fprintf(stderr, "Nb errors corrected : %d\n", nbErrorsCorrected); + fprintf(stderr, "Difference : %d\n", nbErrorsCorrected-nbErrorsIntroduced); std::map<std::string, std::pair<float,std::pair<float,float> > > scores; for (auto & it : counts) - { - // scores[it.first].first = 100.0 * it.second.second / it.second.first; scores[it.first].first = 100.0*pred1Hyp1 / (pred1Hyp1+pred1Hyp0); - } return scores; }