Official code repository for "Drift-diffusion dynamics of hippocampal replay" by Zhongxuan Wu and Xue-Xin Wei.
.
├── ssm/ # Implementation of the switching HMM used in the paper
├── submission/ # Code to reproduce paper figures (to be reorganized during revision)
├── preprocessing_utils.py # Data preprocessing utilities
├── preprocessing_Pfeiffer1D.py # Data preprocessing scipt
├── train_Pfeiffer1D.py # Probabilistic inference on the preprocessed data
├── data_train_shuffle.py # Time-bin + neuron-identity shuffles + runs inference
├── data_train_shuffle_pfs.py # Place-field rotation shuffle + runs inference
├── data_train_simulation_recovery.py # Simulated SWRs (drift-diffusion only) for parameter recovery + runs inference
└── data_train_simulation_full_recovery.py # Simulated SWRs (all basic dynamics) for parameter recovery + runs inference
This study uses data from Pfeiffer and Foster (2015). Please contact Brad Pfeiffer and David Foster to access the analyzed datasets.