Skip to content

Commit

Permalink
moved docstring
Browse files Browse the repository at this point in the history
  • Loading branch information
David committed May 28, 2018
1 parent 712d146 commit 2a98689
Showing 1 changed file with 6 additions and 6 deletions.
12 changes: 6 additions & 6 deletions hpfrec/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,12 +5,6 @@
pd.options.mode.chained_assignment = None

class HPF:
def __init__(self, k=30, a=0.3, a_prime=0.3, b_prime=1.0,
c=0.3, c_prime=0.3, d_prime=1.0, ncores=-1,
stop_crit='train-llk', check_every=10, stop_thr=1e-3,
maxiter=100, reindex=True, random_seed = None,
allow_inconsistent_math=False, verbose=True, full_llk=True,
keep_data=True, save_folder=None, produce_dicts=True):
"""
Hierarchical Poisson Factorization
Expand Down Expand Up @@ -132,6 +126,12 @@ def __init__(self, k=30, a=0.3, a_prime=0.3, b_prime=1.0,
----------
[1] Scalable Recommendation with Hierarchical Poisson Factorization (P. Gopalan, 2015)
"""
def __init__(self, k=30, a=0.3, a_prime=0.3, b_prime=1.0,
c=0.3, c_prime=0.3, d_prime=1.0, ncores=-1,
stop_crit='train-llk', check_every=10, stop_thr=1e-3,
maxiter=100, reindex=True, random_seed = None,
allow_inconsistent_math=False, verbose=True, full_llk=True,
keep_data=True, save_folder=None, produce_dicts=True):

## checking input
assert isinstance(k, int)
Expand Down

0 comments on commit 2a98689

Please sign in to comment.