plot_pre_post_budget_allocation_comparison error with Geo data

[virithavanama]

I running the MMM with 2 channels and 10 geo locations. Model has mean R2 score of 98, r_hat values <1.1
But when running plot_pre_post_budget_allocation_comparison, its giving me below error:

Also plot_media_baseline_contribution_area_plot is giving below error:

[pabloduque0]

@virithavanama can you post the screenshot of the full stack trace in both cases? (Click to expand all frames). That would help us to spot what is going on.

[virithavanama]

plot_pre_post_budget_allocation_comparison:

plot_media_baseline_contribution_area_plot:

[pabloduque0]

Can you share the shapes of:

• mmm.media
• optimal_budget_allocation
• previous_budget_allocation

And the length of mmm.media_names

Thanks!

[virithavanama]

mmm.media -> (195, 2, 10)
optimal_budget_allocation -> (2,)
previous_budget_allocation -> (2,)
mmm.media_names -> (20,)

[pabloduque0]

Did you provide media_names directly?

If so, you only need to provide a list of length n_channels we will add geo suffix/labeling for plots.

[virithavanama]

Thank you, removing the media_names solved the issue!