A fast and versatile implementation of spike-triggered non-negative matrix factorization (STNMF) based on accelerated fast hierarchical alternating least squares (AF-HALS) algorithms.
This Python package allows fast inference of receptive-field subunits from the spiking responses of retinal ganglion cells including methods of hyperparameter tuning.
Described in the paper:
Zapp SJ, Khani MH, Schreyer HM, Sridhar S, Ramakrishna V, Krüppel S, Protti DA, Mietsch M, Karamanlis D, Gollisch T (2024). Accelerated spike-triggered non-negative matrix factorization reveals coordinated ganglion cell subunit mosaics in the primate retina. bioRxiv, 590506. https://doi.org/10.1101/2024.04.22.590506
The documentation is available at https://stnmf.readthedocs.io.
Install using pip from command-line:
pip install stnmfFor feedback and bug reports, please use the Github issue tracker.