Skip to content

Commit 7a51bd3

Browse files
committed
fix zero division error by adding an epsilon
1 parent d85c21d commit 7a51bd3

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

cornac/models/bpr/recom_bpr.pyx

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -196,7 +196,7 @@ class BPR(Recommender, ANNMixin):
196196
user_ids, X.indices, neg_item_ids, X.indptr,
197197
self.u_factors, self.i_factors, self.i_biases)
198198
progress.set_postfix({
199-
"correct": "%.2f%%" % (100.0 * correct / (len(user_ids) - skipped)),
199+
"correct": "%.2f%%" % (100.0 * correct / (len(user_ids) - skipped + 1e-8)),
200200
"skipped": "%.2f%%" % (100.0 * skipped / len(user_ids))
201201
})
202202

cornac/models/bpr/recom_wbpr.pyx

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -135,7 +135,7 @@ class WBPR(BPR):
135135
user_ids, X.indices, X.indices, X.indptr,
136136
self.u_factors, self.i_factors, self.i_biases)
137137
progress.set_postfix({
138-
"correct": "%.2f%%" % (100.0 * correct / (len(user_ids) - skipped)),
138+
"correct": "%.2f%%" % (100.0 * correct / (len(user_ids) - skipped + 1e-8)),
139139
"skipped": "%.2f%%" % (100.0 * skipped / len(user_ids))
140140
})
141141
if self.verbose:

0 commit comments

Comments
 (0)