Skip to content

Tigramite is a python package for causal inference with a focus on time series data. The Tigramite documentation is at

License

Notifications You must be signed in to change notification settings

ZhangY-u/tigramite

 
 

Repository files navigation

Tigramite – Causal inference and causal discovery for time series datasets

Version 5.0

(Python Package)

Github

Documentation

Tutorials

Simple GUI

Overview

Update: Tigramite now has a new CausalEffects class that allows to estimate (conditional) causal effects and mediation based on a causal graph. Have a look at the tutorial.

Further, Tigramite provides several causal discovery methods that can be used under different sets of assumptions. An application always consists of a method and a chosen conditional independence test, e.g. PCMCI together with ParCorr. The following two tables give an overview of the assumptions involved:

Method Assumptions Output
(in addition to Causal Markov Condition and Faithfulness)
PCMCI Causal stationarity, no contemporaneous causal links, no hidden variables Directed lagged links, undirected contemporaneous links (for tau_min=0)
PCMCIplus Causal stationarity, no hidden variables Directed lagged links, directed and undirected contemp. links (Time series CPDAG)
LPCMCI Causal stationarity Time series PAG
Conditional independence test Assumptions
ParCorr univariate variables with continuous, linear Gaussian dependencies
GPDC / GPDCtorch univariate variables with continuous, additive dependencies
CMIknn multivariate variables with continuous, general dependencies
CMIsymb univariate with discrete/categorical dependencies

General Notes

Tigramite is a causal time series analysis python package. It allows to efficiently estimate causal graphs from high-dimensional time series datasets (causal discovery) and to use graphs for robust forecasting and the estimation and prediction of direct, total, and mediated effects. Causal discovery is based on linear as well as non-parametric conditional independence tests applicable to discrete or continuously-valued time series. Also includes functions for high-quality plots of the results. You can find a simple GUI version on https://github.com/stbachinger/TigramiteGui. Please cite the following papers depending on which method you use:

Features

  • high detection power even for large-scale time series datasets
  • flexible conditional independence test statistics adapted to continuously-valued or discrete data, and different assumptions about linear or nonlinear dependencies
  • automatic hyperparameter optimization for most tests
  • parallel computing script based on mpi4py
  • handling of missing values and masks
  • p-value correction and confidence interval estimation
  • causal effect class to non-parametrically estimate (conditional) causal effects and also linear mediated causal effects
  • prediction class based on sklearn models including causal feature selection
  • GUI covering basic functionality on https://github.com/stbachinger/TigramiteGui

Required python packages

  • python=3.7/3.8/3.9
  • numpy==1.21.5
  • scipy==1.8.0
  • numba==0.55.1

Optional packages depending on used functions

  • scikit-learn>=0.21 # Gaussian Process (GP) Regression
  • matplotlib>=3.4.0 # Plotting
  • networkx>=2.4 # Plotting
  • torch>=1.11.0 # GPDC pytorch version (in conda install pytorch)
  • gpytorch>=1.4 # GPDC gpytorch version
  • dcor>=0.5.3 # GPDC distance correlation version

Installation

python setup.py install

This will install tigramite in your path.

To use just the ParCorr, CMIknn, and CMIsymb independence tests, only numpy/numba and scipy are required. For other independence tests more packages are required:

  • GPDC: scikit-learn is required for Gaussian Process regression and dcor for distance correlation

  • GPDCtorch: gpytorch is required for Gaussian Process regression

Note: Due to incompatibility issues between numba and numpy/scipy, we currently freeze their versions.

User Agreement

By downloading TIGRAMITE you agree with the following points: TIGRAMITE is provided without any warranty or conditions of any kind. We assume no responsibility for errors or omissions in the results and interpretations following from application of TIGRAMITE.

You commit to cite above papers in your reports or publications.

License

Copyright (C) 2014-2022 Jakob Runge

See license.txt for full text.

TIGRAMITE is free software; you can redistribute it and/or modify it under the terms of the GNU General Public License as published by the Free Software Foundation; either version 3 of the License, or (at your option) any later version. TIGRAMITE is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for more details.

About

Tigramite is a python package for causal inference with a focus on time series data. The Tigramite documentation is at

Resources

License

Stars

Watchers

Forks

Packages

No packages published

Languages

  • Jupyter Notebook 74.7%
  • Python 25.3%