diff --git a/cornac/models/trirank/recom_trirank.py b/cornac/models/trirank/recom_trirank.py index f4e428f65..427856443 100644 --- 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