search

WillianFuks / tfcausalimpact

Python Causal Impact Implementation Based on Google's R Package. Built using TensorFlow Probability.
Apache License 2.0
610 stars 72 forks source link

Question about output extraction #88

Closed aazz7777 closed 6 months ago

aazz7777 commented 6 months ago

I'm wondering is there a way I can extract the prediction value separately instead of plot.

WillianFuks commented 6 months ago

Hi @aazz7777 ,

Yes, this is possible. The inferences are stored in ci.inferences which is a pandas dataframe with all information extracted from the fitted model.

You also have access to ci.summary_data which relates to the summarization step, such as the cumulative information.

Let me know if this helps.

aazz7777 commented 6 months ago

Thank you !