minor changes in recom_pmf doc

PreferredAI · Oct 12, 2018 · 972f757 · 972f757
1 parent 891aaa9
commit 972f757
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 4 deletions.
diff --git a/cornac/models/pmf/recom_pmf.py b/cornac/models/pmf/recom_pmf.py
@@ -54,12 +54,11 @@ class Pmf(Recommender):
 
     V: csc_matrix, shape (n_items,k)
         The item latent factors, optional initialization via init_params.
-                
+
     References
     ----------
     * Mnih, Andriy, and Ruslan R. Salakhutdinov. Probabilistic matrix factorization. \
     In NIPS, pp. 1257-1264. 2008.
-
     """
 
     def __init__(self, k=5, max_iter=100, learning_rate = 0.001,gamma = 0.9, lamda = 0.001, name = "pmf", variant ='non_linear', trainable = True,init_params = None):

diff --git a/docs/source/models.rst b/docs/source/models.rst
@@ -6,7 +6,7 @@ models
     :members:
 
 Probabilistic Collaborative Representation Learning (PCRL)
------------------------------------------------------------
+------------------------------------------------------------
 .. automodule:: cornac.models.pcrl.recom_pcrl
    :members:
 
@@ -26,7 +26,7 @@ Bayesian Personalized Ranking (BPR)
    :members:
 
 Probabilitic Matrix Factorization (PMF)
-----------------------------------------
+-----------------------------------------
 .. automodule:: cornac.models.pmf.recom_pmf
    :members: