├── .coveragerc ├── .flake8 ├── .gitattributes ├── .github └── workflows │ ├── publish-pypi.yml │ └── run-CI.yml ├── .gitignore ├── LICENSE ├── MANIFEST.in ├── README.md ├── causalimpact ├── __init__.py ├── __version__.py ├── data.py ├── inferences.py ├── main.py ├── misc.py ├── model.py ├── plot.py ├── summary.py └── summary │ └── templates │ ├── report │ └── summary ├── notebooks ├── R │ ├── arma │ │ ├── Rplot.png │ │ └── summary.txt.txt │ ├── arma_data.csv │ ├── basque.csv │ ├── basque │ │ ├── Rplot.png │ │ └── summary.txt.txt │ ├── comparison │ │ ├── Pythonplot.png │ │ ├── Rplot.png │ │ └── summary.txt.txt │ ├── comparison_data.csv │ ├── google │ │ ├── Rplot.png │ │ └── summary.txt.txt │ ├── google_data.csv │ ├── volks │ │ ├── Rplot.png │ │ └── summary.txt.txt │ └── volks_data.csv ├── getting_started.ipynb ├── tfcausal_plot_example.png └── tfcausal_plot_original_example.png ├── pytest.ini ├── scripts └── build_wheels.sh ├── setup.cfg ├── setup.py ├── stubs └── tensorflow_probability │ ├── __init__.pyi │ └── sts │ └── __init__.pyi ├── test-requirements.txt ├── tests ├── conftest.py ├── fixtures │ ├── arma_data.csv │ ├── arma_sparse_reg.csv │ ├── basque.csv │ ├── btc.csv │ ├── comparison_data.csv │ ├── google_data.csv │ ├── season_data.csv │ ├── test_output_summary_1 │ ├── test_output_summary_single_digit │ ├── test_report_summary_1 │ ├── test_report_summary_2 │ ├── test_report_summary_3 │ ├── test_report_summary_4 │ ├── test_report_summary_single_digit │ └── volks_data.csv ├── test_data.py ├── test_inferences.py ├── test_main.py ├── test_misc.py ├── test_model.py ├── test_plot.py └── test_summary.py └── tox.ini /.coveragerc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WillianFuks/tfcausalimpact/HEAD/.coveragerc -------------------------------------------------------------------------------- /.flake8: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WillianFuks/tfcausalimpact/HEAD/.flake8 -------------------------------------------------------------------------------- /.gitattributes: -------------------------------------------------------------------------------- 1 | notebooks/* linguist-documentation 2 | -------------------------------------------------------------------------------- /.github/workflows/publish-pypi.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WillianFuks/tfcausalimpact/HEAD/.github/workflows/publish-pypi.yml -------------------------------------------------------------------------------- /.github/workflows/run-CI.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WillianFuks/tfcausalimpact/HEAD/.github/workflows/run-CI.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WillianFuks/tfcausalimpact/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WillianFuks/tfcausalimpact/HEAD/LICENSE -------------------------------------------------------------------------------- /MANIFEST.in: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WillianFuks/tfcausalimpact/HEAD/MANIFEST.in -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WillianFuks/tfcausalimpact/HEAD/README.md -------------------------------------------------------------------------------- /causalimpact/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WillianFuks/tfcausalimpact/HEAD/causalimpact/__init__.py -------------------------------------------------------------------------------- /causalimpact/__version__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WillianFuks/tfcausalimpact/HEAD/causalimpact/__version__.py -------------------------------------------------------------------------------- /causalimpact/data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WillianFuks/tfcausalimpact/HEAD/causalimpact/data.py -------------------------------------------------------------------------------- /causalimpact/inferences.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WillianFuks/tfcausalimpact/HEAD/causalimpact/inferences.py -------------------------------------------------------------------------------- /causalimpact/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WillianFuks/tfcausalimpact/HEAD/causalimpact/main.py -------------------------------------------------------------------------------- /causalimpact/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WillianFuks/tfcausalimpact/HEAD/causalimpact/misc.py -------------------------------------------------------------------------------- /causalimpact/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WillianFuks/tfcausalimpact/HEAD/causalimpact/model.py -------------------------------------------------------------------------------- /causalimpact/plot.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WillianFuks/tfcausalimpact/HEAD/causalimpact/plot.py -------------------------------------------------------------------------------- /causalimpact/summary.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WillianFuks/tfcausalimpact/HEAD/causalimpact/summary.py -------------------------------------------------------------------------------- /causalimpact/summary/templates/report: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WillianFuks/tfcausalimpact/HEAD/causalimpact/summary/templates/report -------------------------------------------------------------------------------- /causalimpact/summary/templates/summary: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WillianFuks/tfcausalimpact/HEAD/causalimpact/summary/templates/summary -------------------------------------------------------------------------------- /notebooks/R/arma/Rplot.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WillianFuks/tfcausalimpact/HEAD/notebooks/R/arma/Rplot.png -------------------------------------------------------------------------------- /notebooks/R/arma/summary.txt.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WillianFuks/tfcausalimpact/HEAD/notebooks/R/arma/summary.txt.txt -------------------------------------------------------------------------------- /notebooks/R/arma_data.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WillianFuks/tfcausalimpact/HEAD/notebooks/R/arma_data.csv -------------------------------------------------------------------------------- /notebooks/R/basque.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WillianFuks/tfcausalimpact/HEAD/notebooks/R/basque.csv -------------------------------------------------------------------------------- /notebooks/R/basque/Rplot.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WillianFuks/tfcausalimpact/HEAD/notebooks/R/basque/Rplot.png -------------------------------------------------------------------------------- /notebooks/R/basque/summary.txt.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WillianFuks/tfcausalimpact/HEAD/notebooks/R/basque/summary.txt.txt -------------------------------------------------------------------------------- /notebooks/R/comparison/Pythonplot.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WillianFuks/tfcausalimpact/HEAD/notebooks/R/comparison/Pythonplot.png -------------------------------------------------------------------------------- /notebooks/R/comparison/Rplot.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WillianFuks/tfcausalimpact/HEAD/notebooks/R/comparison/Rplot.png -------------------------------------------------------------------------------- /notebooks/R/comparison/summary.txt.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WillianFuks/tfcausalimpact/HEAD/notebooks/R/comparison/summary.txt.txt -------------------------------------------------------------------------------- /notebooks/R/comparison_data.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WillianFuks/tfcausalimpact/HEAD/notebooks/R/comparison_data.csv -------------------------------------------------------------------------------- /notebooks/R/google/Rplot.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WillianFuks/tfcausalimpact/HEAD/notebooks/R/google/Rplot.png -------------------------------------------------------------------------------- /notebooks/R/google/summary.txt.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WillianFuks/tfcausalimpact/HEAD/notebooks/R/google/summary.txt.txt -------------------------------------------------------------------------------- /notebooks/R/google_data.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WillianFuks/tfcausalimpact/HEAD/notebooks/R/google_data.csv -------------------------------------------------------------------------------- /notebooks/R/volks/Rplot.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WillianFuks/tfcausalimpact/HEAD/notebooks/R/volks/Rplot.png -------------------------------------------------------------------------------- /notebooks/R/volks/summary.txt.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WillianFuks/tfcausalimpact/HEAD/notebooks/R/volks/summary.txt.txt -------------------------------------------------------------------------------- /notebooks/R/volks_data.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WillianFuks/tfcausalimpact/HEAD/notebooks/R/volks_data.csv -------------------------------------------------------------------------------- /notebooks/getting_started.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WillianFuks/tfcausalimpact/HEAD/notebooks/getting_started.ipynb -------------------------------------------------------------------------------- /notebooks/tfcausal_plot_example.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WillianFuks/tfcausalimpact/HEAD/notebooks/tfcausal_plot_example.png -------------------------------------------------------------------------------- /notebooks/tfcausal_plot_original_example.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WillianFuks/tfcausalimpact/HEAD/notebooks/tfcausal_plot_original_example.png -------------------------------------------------------------------------------- /pytest.ini: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WillianFuks/tfcausalimpact/HEAD/pytest.ini -------------------------------------------------------------------------------- /scripts/build_wheels.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WillianFuks/tfcausalimpact/HEAD/scripts/build_wheels.sh -------------------------------------------------------------------------------- /setup.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WillianFuks/tfcausalimpact/HEAD/setup.cfg -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WillianFuks/tfcausalimpact/HEAD/setup.py -------------------------------------------------------------------------------- /stubs/tensorflow_probability/__init__.pyi: -------------------------------------------------------------------------------- 1 | from .sts import LocalLevel # noqa: F401 2 | -------------------------------------------------------------------------------- /stubs/tensorflow_probability/sts/__init__.pyi: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WillianFuks/tfcausalimpact/HEAD/stubs/tensorflow_probability/sts/__init__.pyi -------------------------------------------------------------------------------- /test-requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WillianFuks/tfcausalimpact/HEAD/test-requirements.txt -------------------------------------------------------------------------------- /tests/conftest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WillianFuks/tfcausalimpact/HEAD/tests/conftest.py -------------------------------------------------------------------------------- /tests/fixtures/arma_data.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WillianFuks/tfcausalimpact/HEAD/tests/fixtures/arma_data.csv -------------------------------------------------------------------------------- /tests/fixtures/arma_sparse_reg.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WillianFuks/tfcausalimpact/HEAD/tests/fixtures/arma_sparse_reg.csv -------------------------------------------------------------------------------- /tests/fixtures/basque.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WillianFuks/tfcausalimpact/HEAD/tests/fixtures/basque.csv -------------------------------------------------------------------------------- /tests/fixtures/btc.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WillianFuks/tfcausalimpact/HEAD/tests/fixtures/btc.csv -------------------------------------------------------------------------------- /tests/fixtures/comparison_data.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WillianFuks/tfcausalimpact/HEAD/tests/fixtures/comparison_data.csv -------------------------------------------------------------------------------- /tests/fixtures/google_data.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WillianFuks/tfcausalimpact/HEAD/tests/fixtures/google_data.csv -------------------------------------------------------------------------------- /tests/fixtures/season_data.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WillianFuks/tfcausalimpact/HEAD/tests/fixtures/season_data.csv -------------------------------------------------------------------------------- /tests/fixtures/test_output_summary_1: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WillianFuks/tfcausalimpact/HEAD/tests/fixtures/test_output_summary_1 -------------------------------------------------------------------------------- /tests/fixtures/test_output_summary_single_digit: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WillianFuks/tfcausalimpact/HEAD/tests/fixtures/test_output_summary_single_digit -------------------------------------------------------------------------------- /tests/fixtures/test_report_summary_1: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WillianFuks/tfcausalimpact/HEAD/tests/fixtures/test_report_summary_1 -------------------------------------------------------------------------------- /tests/fixtures/test_report_summary_2: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WillianFuks/tfcausalimpact/HEAD/tests/fixtures/test_report_summary_2 -------------------------------------------------------------------------------- /tests/fixtures/test_report_summary_3: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WillianFuks/tfcausalimpact/HEAD/tests/fixtures/test_report_summary_3 -------------------------------------------------------------------------------- /tests/fixtures/test_report_summary_4: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WillianFuks/tfcausalimpact/HEAD/tests/fixtures/test_report_summary_4 -------------------------------------------------------------------------------- /tests/fixtures/test_report_summary_single_digit: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WillianFuks/tfcausalimpact/HEAD/tests/fixtures/test_report_summary_single_digit -------------------------------------------------------------------------------- /tests/fixtures/volks_data.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WillianFuks/tfcausalimpact/HEAD/tests/fixtures/volks_data.csv -------------------------------------------------------------------------------- /tests/test_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WillianFuks/tfcausalimpact/HEAD/tests/test_data.py -------------------------------------------------------------------------------- /tests/test_inferences.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WillianFuks/tfcausalimpact/HEAD/tests/test_inferences.py -------------------------------------------------------------------------------- /tests/test_main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WillianFuks/tfcausalimpact/HEAD/tests/test_main.py -------------------------------------------------------------------------------- /tests/test_misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WillianFuks/tfcausalimpact/HEAD/tests/test_misc.py -------------------------------------------------------------------------------- /tests/test_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WillianFuks/tfcausalimpact/HEAD/tests/test_model.py -------------------------------------------------------------------------------- /tests/test_plot.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WillianFuks/tfcausalimpact/HEAD/tests/test_plot.py -------------------------------------------------------------------------------- /tests/test_summary.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WillianFuks/tfcausalimpact/HEAD/tests/test_summary.py -------------------------------------------------------------------------------- /tox.ini: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WillianFuks/tfcausalimpact/HEAD/tox.ini --------------------------------------------------------------------------------