pymc-labs / pymc-marketing

Bayesian marketing toolbox in PyMC. Media Mix (MMM), customer lifetime value (CLV), buy-till-you-die (BTYD) models and more.
https://www.pymc-marketing.io/
Apache License 2.0
668 stars 184 forks source link

Add `BetaGeoModel` example and corresponding tests using the CDNOW dataset #100

Open larryshamalama opened 1 year ago

ColtAllen commented 1 year ago

I think it'd be worthwhile to add a dataset utility function for CDNOW. Practically every research paper I've read on continuous, non-contractual models use CDNOW for benchmarking. Reproducing research results is the best way to proceed with testing in my mind, and there are results based on CDNOW available for many other models.

I can create a PR for this.

ColtAllen commented 1 year ago

I've added a pytest fixture for the CDNOW dataset in https://github.com/pymc-labs/pymc-marketing/pull/164 that can be used for this.