[ICML 2024] Multi-Region Markovian Gaussian Process: An Efficient Method to Discover Directional Communications Across Multiple Brain Regions
Weihan Li, Chengrui Li, Yule Wang, and Anqi Wu
From the current directory, run
pip install -r requirements.txt
Our code is highly dependent on the SSM package. To install SSM, run
git clone https://github.com/lindermanlab/ssm
cd ssm
pip install numpy cython
pip install -e .
demo.ipynb and demo_r2.ipynb are two step-by-step tutorials that learn an MRMGP from a demo dataset and evaluate via log-likelihood on held-out neurons and R square on predictions.