csv files for all the plot functions

[virithavanama]

Hi Team,

Can you please provide the way to generate the CSV files for response curve plots and actual vs predicted plots?

[pabloduque0]

@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.

[virithavanama]

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

[shekharkhandelwal1983]

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.

[michevan]

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.