lightweight_mmm
lightweight_mmm copied to clipboard
add got incompatible shapes for broadcasting: (95,), (90,).
I am getting this error when trying to predict. size of media_test_data is (5,3)
I got the exact same issue, seems as if the bug occurs due to a change in any dependency
Did the issue get fixed for you ? I am using below libraries lightweight-mmm numpyro
You can resolve it yourself by forcefully installing an older version of numpyro, i.e, run !pip numpyro == 0.13.2
They just released 0.14.0 today
also install an older version of jax. 'jax.extend.linear_util' was removed in jax after 0.4.23 (currently in 0.4.25) which is called by numpyro.
As a temporary fix, you can do
pip install --upgrade git+https://github.com/google/lightweight_mmm.git