One-hot Generalized Linear Model for Switching Brain State Discovery [ICLR 2024]

Chengrui Li, Soon Ho Kim, Chris Rodgers, Hannah Choi, and Anqi Wu

[paper] [arXiv] [slides] [video] [poster] [文章]

1 Installation

From the current directory (which includes this README.md), run

cd hmmglm
pip install -e .

2 Tutorial

hmmglm/hmmglm/deprecated/demo.ipynb is a step-by-step tutorial that run naive HMM-GLM, Gaussian HMM-GLM, and one-hot HMM-GLM on a synthetic dataset.

experiments/synthetic.ipynb is a step-by-step tutorial that run one-hot HMM-GLM (version 0.1.0) on a synthetic dataset.

experiments/pfc6.ipynb is a step-by-step tutorial that run one-hot HMM-GLM (version 0.1.0) on the CNCRS pfc-6 dataset.