You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Update docs/source/notebooks/bass/bass_example.ipynb to showcase the new BassModel class.
Motivation
The current notebook uses create_bass_model directly. Since the refactor (#2584) added the BassModel(ModelBuilder) class as the recommended entry point, the notebook should demonstrate the class-based workflow.
Sections to update
Import — Use from pymc_marketing.bass import BassModel as primary entry point
Model creation — BassModel(model_config=...) instead of separate priors dict
Fitting — model.fit(data=...) with data in various formats (array, DataFrame, xr.Dataset)
Posterior analysis — Use model.idata directly with ArviZ
Posterior predictive — model.sample_posterior_predictive(X=future) for forecasting
Update
docs/source/notebooks/bass/bass_example.ipynbto showcase the newBassModelclass.Motivation
The current notebook uses
create_bass_modeldirectly. Since the refactor (#2584) added theBassModel(ModelBuilder)class as the recommended entry point, the notebook should demonstrate the class-based workflow.Sections to update
from pymc_marketing.bass import BassModelas primary entry pointBassModel(model_config=...)instead of separate priors dictmodel.fit(data=...)with data in various formats (array, DataFrame, xr.Dataset)model.idatadirectly with ArviZmodel.sample_posterior_predictive(X=future)for forecastingmodel.save()/BassModel.load()round-tripmodel.plot_*methods (once # is complete)pymc_marketing.mlflow.autolog(log_bass=True)(once MLflow integration for BassModel #2588 is complete)Notes
pm.samplecalls withmodel.fitwhere possiblecreate_bass_modelfunction should still be mentioned as a lower-level alternative