I am having trouble figuring out what data should be used for the variable, optimal_allocation_per_timeunit, in the plot_response_curve method. From the documentation, it says optimal_allocation_per_timeunit "can be obtained by running the optimization provided by LightweightMMM."
But I'm not sure exactly which output from running optimize_media.find_optimal_budgets I should use?