Skip to content

Commit

Permalink
docstrings
Browse files Browse the repository at this point in the history
  • Loading branch information
ColtAllen committed Oct 20, 2023
1 parent c8e9fa1 commit 6c5af0b
Showing 1 changed file with 2 additions and 5 deletions.
7 changes: 2 additions & 5 deletions pymc_marketing/clv/models/pareto_nbd.py
Original file line number Diff line number Diff line change
Expand Up @@ -104,10 +104,8 @@ class ParetoNBDModel(CLVModel):
model.fit(fit_method='map')
print(model.fit_summary())
# For more informative predictions and reliable performance on smaller datasets,
# Use 'mcmc' or 'slice' to estimate full posterior distributions.
# 'slice' is usually faster than 'mcmc' for this model.
model.fit(fit_method='slice')
# Use 'mcmc' for more informative predictions and reliable performance on smaller datasets
model.fit(fit_method='mcmc')
print(model.fit_summary())
# Predict number of purchases for customers over the next 10 time periods
Expand Down Expand Up @@ -303,7 +301,6 @@ def fit(self, fit_method: str = "map", **kwargs): # type: ignore
Method used to fit the model. Options are:
* "map": Posterior point estimates via Maximum a Posteriori (default)
* "mcmc": Full posterior distributions via No U-Turn Sampler (NUTS)
* "slice": Full posterior distributions via Slice Sampling.
kwargs:
Other keyword arguments passed to the underlying PyMC routines
"""
Expand Down

0 comments on commit 6c5af0b

Please sign in to comment.