ColtAllen / btyd

Buy Till You Die and Customer Lifetime Value statistical models in Python.
https://btyd.readthedocs.io/
Apache License 2.0
114 stars 9 forks source link

Add Modified BG/NBD Model #66

Closed ColtAllen closed 2 years ago

ColtAllen commented 2 years ago

This PR adds the Modified Beta-Geometric/Negative Binomial Distribution model, as well as a few API enhancements. The main difference between the modified BG/NBD model and the original is that the original assumes all non-repeat customers are still alive - this can be a valid assumption depending on the use case, but could also grossly inflate the total number of expected transactions for the customer population, particularly when modeling over longer time periods. The modified BG/NBD is a good alternative to consider if these issues are being encountered with the original model.