optimized spend - limit channel budget of optimal spend

[ohad-monday]

in the demo example, let's say i want channel_0 to stay in the same level (200) of the spend for some reason.
optimization with default lower/upper limits:

I tried using the lower/upper bounds and limit channel_0 with (lower/upper limits are 0 for channel_0):
import jax.numpy as jnp
upper = jnp.asarray([0,0.9,0.9])
lower = jnp.asarray([0,0.9,0.9])

solution, kpi_without_optim, previous_media_allocation = optimize_media.find_optimal_budgets(
n_time_periods=n_time_periods,
media_mix_model=mmm,
extra_features=extra_features_scaler.transform(extra_features_test)[:n_time_periods],
budget=budget,
prices=prices,
bounds_lower_pct = lower,
bounds_upper_pct = upper,
media_scaler=media_scaler,
target_scaler=target_scaler,
seed=SEED)

optimization with custom lower/upper limits, show lower decrease in but still a decrease (150) but i want it to stay the same:

am i missing something here?

[michevan]

Hi! I just tried replicating this using the mock data in the national-level example colab, and got a different result. Are you still seeing this issue?

[ohad-monday]

yes, still same issue.
i'm using a docker image with lightweight_mmm-0.1.5 - i will try maybe to update to a newer version

[michevan]

sounds good, let us know if that helps! We're up to v0.1.7 now as of today

[ohad-monday]

it works! with an updated version

[michevan]

great to hear, thank you for the update!