Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Implement pre-built BG/NBD/BetaGeoFitter model #39

Closed
larryshamalama opened this issue Jun 14, 2022 · 0 comments · Fixed by #67
Closed

Implement pre-built BG/NBD/BetaGeoFitter model #39

larryshamalama opened this issue Jun 14, 2022 · 0 comments · Fixed by #67
Assignees
Labels
CLV enhancement New feature or request
Milestone

Comments

@larryshamalama
Copy link
Contributor

larryshamalama commented Jun 14, 2022

It would be good to add a BetaGeoFitter function that returns a ContNonContract with some default priors. A signature that resembles what is provided in the lifetimes package would be a good idea. Something along the lines of the following snipet.

def BetaGeoFitter(name, a, b, r, alpha, T, T0, *, observed, **kwargs):
    p = pm.Beta(f"{name}_beta", a, b, size=size, shape=shape)
    lam = pm.Gamma(f"{name}_gamma", r, 1/alpha, size=size, shape=shape)
    return ContNonContract(name, lam, p, T, T0, size=size, shape=shape, **kwargs)

We should also add some useful summary stats / plots. If they are not specific to the BG/NBD the better!

@ricardoV94 ricardoV94 added the CLV label Jun 14, 2022
@ricardoV94 ricardoV94 added this to the Alpha release milestone Jun 16, 2022
@ricardoV94 ricardoV94 added the enhancement New feature or request label Jun 16, 2022
@ricardoV94 ricardoV94 changed the title Add BetaGeoFitter default functionality akin to lifetimes Implement pre-built BG/NBD/BetaGeoFitter model Jun 16, 2022
@ricardoV94 ricardoV94 added enhancement New feature or request and removed enhancement New feature or request labels Jun 16, 2022
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
CLV enhancement New feature or request
Projects
None yet
Development

Successfully merging a pull request may close this issue.

2 participants