Update TriRank documentation (#619)

PreferredAI · May 20, 2024 · 55e7f2e · 55e7f2e
1 parent f8bbdf6
commit 55e7f2e
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/cornac/models/trirank/recom_trirank.py b/cornac/models/trirank/recom_trirank.py
@@ -69,12 +69,12 @@ class TriRank(Recommender):
  R: csr_matrix, shape (n_users, n_items)
  The symmetric normalized of edge weight matrix of user-item relation, optional initialization via init_params
 
- X: csr_matrix, shape (n_users, n_aspects)
- The symmetric normalized of edge weight matrix of user-aspect relation, optional initialization via init_params
-
- Y: csr_matrix, shape (n_items, n_aspects)
+ X: csr_matrix, shape (n_items, n_aspects)
  The symmetric normalized of edge weight matrix of item-aspect relation, optional initialization via init_params
 
+ Y: csr_matrix, shape (n_users, n_aspects)
+ The symmetric normalized of edge weight matrix of user-aspect relation, optional initialization via init_params
+
  p: ndarray, shape (n_items,)
  Initialized item weights, optional initialization via init_params