Might want to check against the default implementation in MMM Example Notebook — pymc-marketing 0.1.1 documentation