diff --git a/cornac/models/companion/recom_companion.pyx b/cornac/models/companion/recom_companion.pyx index c2bba091..32c9047b 100644 --- a/cornac/models/companion/recom_companion.pyx +++ b/cornac/models/companion/recom_companion.pyx @@ -497,7 +497,7 @@ class Companion(Recommender): progress.set_postfix({ "loss": "%.2f" % (loss / 3 / self.n_element_samples), "bpr_loss": "%.2f" % (bpr_loss / self.n_bpr_samples), - "correct": "%.2f%%" % (100.0 * correct / (self.n_bpr_samples - skipped)), + "correct": "%.2f%%" % (100.0 * correct / (self.n_bpr_samples - skipped + 1e-8)), "skipped": "%.2f%%" % (100.0 * skipped / self.n_bpr_samples), }) diff --git a/cornac/models/comparer/recom_comparer_sub.pyx b/cornac/models/comparer/recom_comparer_sub.pyx index 9b2bf661..5d5d7b4d 100644 --- a/cornac/models/comparer/recom_comparer_sub.pyx +++ b/cornac/models/comparer/recom_comparer_sub.pyx @@ -472,7 +472,7 @@ class ComparERSub(MTER): progress.set_postfix({ "loss": "%.2f" % (loss / 3 / self.n_element_samples), "bpr_loss": "%.2f" % (bpr_loss / self.n_bpr_samples), - "correct": "%.2f%%" % (100.0 * correct / (self.n_bpr_samples - skipped)), + "correct": "%.2f%%" % (100.0 * correct / (self.n_bpr_samples - skipped + 1e-8)), "skipped": "%.2f%%" % (100.0 * skipped / self.n_bpr_samples) }) diff --git a/cornac/models/lrppm/recom_lrppm.pyx b/cornac/models/lrppm/recom_lrppm.pyx index 236da8ac..3e87c12a 100644 --- a/cornac/models/lrppm/recom_lrppm.pyx +++ b/cornac/models/lrppm/recom_lrppm.pyx @@ -331,7 +331,7 @@ class LRPPM(Recommender): "loss": "%.2f" % (loss / self.n_samples), "ranking_loss": "%.2f" % (ranking_loss / (self.n_ranking_samples - skipped)), "r_loss": "%.2f" % (r_loss / (self.n_ranking_samples - skipped)), - "correct": "%.2f%%" % (100.0 * correct / (self.n_ranking_samples - skipped)), + "correct": "%.2f%%" % (100.0 * correct / (self.n_ranking_samples - skipped + 1e-8)), "skipped": "%.2f%%" % (100.0 * skipped / self.n_ranking_samples) }) if (