Skip to content

Commit 81e88df

Browse files
Fix mixmatch in opinion handling (#685)
1 parent a995589 commit 81e88df

1 file changed

Lines changed: 5 additions & 5 deletions

File tree

cornac/models/companion/recom_companion.pyx

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -855,7 +855,7 @@ class Companion(Recommender):
855855
s = 1
856856
if user_item_aspect_neg_opinion_dict.my_map.find(get_key3(get_key(u_idx, i_idx), a_idx, o_jdx)) != user_item_aspect_neg_opinion_dict.my_map.end():
857857
i_score = YN[idx]
858-
j_score = user_item_aspect_pos_opinion_dict.my_map[get_key3(get_key(u_idx, i_idx), a_idx, o_jdx)]
858+
j_score = user_item_aspect_neg_opinion_dict.my_map[get_key3(get_key(u_idx, i_idx), a_idx, o_jdx)]
859859
if i_score == j_score:
860860
skipped_uiaon += 1
861861
continue
@@ -879,10 +879,10 @@ class Companion(Recommender):
879879
for j in range(n_aspect_factors):
880880
for i in range(n_user_factors):
881881
del_g3[i, j, k] -= del_bpr_uiaon * U[u_idx, i] * A[a_idx, j] * uiaon_ij
882-
del_u[u_idx, i] -= del_bpr_uiaon * G2[i, j, k] * A[a_idx, j] * uiaon_ij
883-
del_a[a_idx, j] -= del_bpr_uiaon * G2[i, j, k] * U[u_idx, i] * uiaon_ij
884-
del_o[o_idx, k] -= del_bpr_uiaon * G2[i, j, k] * U[u_idx, i] * A[a_idx, j]
885-
del_o[o_jdx, k] += del_bpr_uiaon * G2[i, j, k] * U[u_idx, i] * A[a_idx, j]
882+
del_u[u_idx, i] -= del_bpr_uiaon * G3[i, j, k] * A[a_idx, j] * uiaon_ij
883+
del_a[a_idx, j] -= del_bpr_uiaon * G3[i, j, k] * U[u_idx, i] * uiaon_ij
884+
del_o[o_idx, k] -= del_bpr_uiaon * G3[i, j, k] * U[u_idx, i] * A[a_idx, j]
885+
del_o[o_jdx, k] += del_bpr_uiaon * G3[i, j, k] * U[u_idx, i] * A[a_idx, j]
886886
for i in range(n_item_factors):
887887
del_g3[n_user_factors + i, j, k] -= del_bpr_uiaon * I[i_idx, i] * A[a_idx, j] * uiaon_ij
888888
del_i[i_idx, i] -= del_bpr_uiaon * G3[n_user_factors + i, j, k] * A[a_idx, j] * uiaon_ij

0 commit comments

Comments
 (0)