csv files for all the plot functions about lightweight_mmm HOT 4 OPEN

@virithavanama Thanks for requesting this!

CSV might not be the format we would return it directly but either an array or DataFrame could be, which is easily converted to CSV.

We will look into it.

from lightweight_mmm.

can we get the dataframe for recreating the 'plot_pre_post_budget_allocation_comparison' and 'plot_model_fit' plots

from lightweight_mmm.

Hi @pabloduque0 , just checking on this request. Even I would like to know how can we get data in any format (csv/array etc) for all model outputs.

from lightweight_mmm.

hi virithavanama and shekharkhandelwal1983!

For the plot_model_fit plot, both the target curve and the posterior predictions are produced by lines 662-666 of plot.py, so you can just replicate those in a Colab to get the data.

For the plot_pre_post_budget_allocation_comparison, the data are passed as inputs to those functions so you should already have them, but lines 843-845 convert them to fractions if that's easier to work with.

from lightweight_mmm.