Add Bass Diffusion Model

[wd60622]

Description

Adding Bass Diffusion Model

Related Issue

• Closes Add Bass Diffusion Model #1126
• Related to #

Checklist

• Checked that the pre-commit linting/style checks pass
• Included tests that prove the fix is effective or that the new feature works
• Added necessary documentation (docstrings and/or example notebooks)
• If you are a pro: each commit corresponds to a relevant logical change

Modules affected

• MMM
• CLV
• Customer Choice
• Product Development

Type of change

• New feature / enhancement
• Bug fix
• Documentation
• Maintenance
• Other (please specify):

---

📚 Documentation preview 📚: https://pymc-marketing--1328.org.readthedocs.build/en/1328/

[wd60622]

Here is a crude implementation of the Bass model. Feel free to take over @juanitorduz

I forget why I didn't wrap it in the ModelBuilder. Maybe I was just trying it out and familiarizing myself with the model. Maybe it will be straight forward but I remember having some concern.

As for the magnitude of m compared to the other parameters and NUTS, maybe a scaling constant can be used in the model. You might have some more ideas.

model = Bass(m_scaling=5000)

# Under the hood
m = m_scaling * Prior.create_variable(...)

[codecov]

Codecov Report

Attention: Patch coverage is 0% with 38 lines in your changes missing coverage. Please review.

Project coverage is 94.58%. Comparing base (f80581b) to head (5172c61).

Files with missing lines	Patch %	Lines
pymc_marketing/bass.py	0.00%	38 Missing ⚠️

Additional details and impacted files

@@            Coverage Diff             @@
##             main    #1328      +/-   ##
==========================================
- Coverage   95.31%   94.58%   -0.74%     
==========================================
  Files          47       48       +1     
  Lines        4912     4950      +38     
==========================================
  Hits         4682     4682              
- Misses        230      268      +38     

☔ View full report in Codecov by Sentry.
📢 Have feedback on the report? Share it here.

[juanitorduz]

Thanks @wd60622 ! I will familiarize myself with the model and push it forward 🙌

[wd60622]

Sounds good! Thanks

I had the idea we could parameterize it as days/weeks/months from product launch making pooling across dims / products easier. I feel like that would be an interesting insight.

[wd60622]

Visual produced from the main block: