Get contribution from predict

[Citronelol]

trafficstars

Hello there,

First of all, huge thanks for this awesome project.

I was wondering : would it be possible to extract the contribution of each media channel when using the predict() method ?

I am aware of this method: https://github.com/google/lightweight_mmm/blob/main/lightweight_mmm/plot.py#L144 but it seems that it returns the contribution on the train time series. How would one use it to decompose predictions ?

Many thanks in advance !