UnitMatchPy and DeepUnitMatch
This repository contains:
- UnitMatchPy: automatic matching of neurons across sessions (Python package).
- DeepUnitMatch: DeepUnitMatch pipeline and demos (see preprint below).
References
- UnitMatch: https://www.nature.com/articles/s41592-024-02440-1
- DeepUnitMatch preprint: https://www.biorxiv.org/content/10.64898/2026.01.30.702777v1
Versions
- UnitMatchPy version:
3.5.0(frompyproject.toml) - DeepUnitMatch: code lives under
DeepUnitMatch/(see DeepUnitMatch (DUM) section)
System requirements
Both UnitMatchPy and DeepUnitMatch can be ran on a standard computer, with sufficient RAM (>32GB for large datasets). This software is supported for Windows and macOS, and has been tested on Windows 11.
Installation
pip install (including pip install -e .) installs into whatever Python environment your pip points to (system Python, a conda env, or a virtualenv). pip does not create or name environments.
We recommend using Anaconda/Miniconda (conda) to create an isolated environment first:
# Create a new environment (pick any name you like; example: UMPy)
conda create -n UMPy python=3.11 pip #(press y when prompted)
conda activate UMPy
Then install using pip (options below). This step should take less than few minutes.
Option A: Install the released package (PyPI)
pip install UnitMatchPy
Optional extras (heavier dependencies used by some notebooks and integration with SpikeInterface):
pip install "UnitMatchPy[full,notebooks]"
Option B: Install a local, editable copy (for development / modified code)
First, open a terminal and navigate to this folder (the one containing pyproject.toml). The pip install -e command must be run from here:
# Windows (PowerShell) - note; you may have to give writing access to the specific CONDA environment via Windows Security → Virus & threat protection → Ransomware protection → Manage ransomware protection. For example, if the below steps result in "could not create 'UnitMatchPy.egg-info'"
Look at Controlled folder access:
cd $HOME\Documents\GitHub\UnitMatch\UnitMatchPy
# macOS / Linux
cd ~/Documents/GitHub/UnitMatch/UnitMatchPy
pip install -e .
Optional extras (e.g. if you'd like to run the notebooks or integrate with SpikeInterface):
pip install -e ".[full,notebooks]"
Demo notebooks
All demo notebooks are in Demo Notebooks/.
Run UnitMatchPy
To run UnitMatchPy, standard spike sorting data is needed (channel positions and extracted raw waveforms for each unit). Waveforms can be extracted externally (e.g. BombCell) or using the demo notebooks:
Demo Notebooks/extract_raw_data_demo.ipynb(compressed.cbin/.chor raw)Demo Notebooks/extract_raw_data_demo_open_ephys.ipynb(Open Ephys)Demo Notebooks/UMPy_spike_interface_demo.ipynb(SpikeInterface workflow)
Example notebooks:
Demo Notebooks/UMPy_example.ipynb(recommended starting point)Demo Notebooks/UMPy_example_detailed.ipynb(more modular / advanced)
The GUI is an optional step to curate and explore UnitMatch outputs; see Demo Notebooks/GUI_Reference_Guide.md for usage tips and shortcuts.
Run DeepUnitMatch
To try the DeepUnitMatch version, start with:
Demo Notebooks/DeepUnitMatch.ipynb
It should take just a few minutes to run on a standard PC.
If you want to train / fine-tune a model on your own data, see:
Demo Notebooks/DUM_training.ipynb
Important note: DeepUnitMatch current trained model is for Npix 2.0 4-shank only. You will need to train a new model with your own data if you have any other type of probe. (For example, Mouse2 from the figshare data is a Npix 1 dataset, you'll notice the trained model won't give good results on this mouse for that reason.)
4 maintainers
Metadata
4 maintainers
Release files for UnitMatchPy 3.5.0
For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.
Source distribution (sdist)
| File | Size | Uploaded | |
|---|---|---|---|
| unitmatchpy-3.5.0.tar.gz | 188.8 kB | Details |
Built distribution (wheel)
| File | Interpreter | ABI | Platform | Reset |
|---|---|---|---|---|
| unitmatchpy-3.5.0-py3-none-any.whl | Python 3 | none | any | Details |
Total release size: 681.3 kB
Release files / unitmatchpy-3.5.0.tar.gz
| Download URL | unitmatchpy-3.5.0.tar.gz |
|---|---|
| Size | 188.8 kB |
| Tags | Source |
|
SHA-256 checksum How to use checksums |
b7839549ae4f7255f96575d2c4302bad1252f7d5de0e3f2c5ea88d627ec81a02
|
|
BLAKE2b-256 checksum How to use checksums |
f89475b00992e3b83fb3036f9f7946a0078cba7106bd2cb314218bde8242db7a
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
Yes |
| Uploaded via |
twine/6.1.0 CPython/3.13.14
|
Provenance
Provenance describes where a file came from. On PyPI, provenance is shared via attestations, which provide a verifiable record of the build or publishing details. View details, limitations and caveats.
PyPI Publish Attestation
PyPI verified that this artifact, at this checksum, originated from the publisher listed below.
Signed by GitHub Actions, verified by PyPI on Jul 20, 2026.
Transparency logRelease files / unitmatchpy-3.5.0-py3-none-any.whl
| Download URL | unitmatchpy-3.5.0-py3-none-any.whl |
|---|---|
| Size | 492.5 kB |
| Tags | Python 3 |
|
SHA-256 checksum How to use checksums |
cb1b4f47717fbfdd81d3709536e6d2e75a139fd3b4baf6d766e73acbcc8df588
|
|
BLAKE2b-256 checksum How to use checksums |
5c25e298c68df5fa7c3ee5320ee43f0acd0014a62f18b57f400acaed25bfa5d6
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
Yes |
| Uploaded via |
twine/6.1.0 CPython/3.13.14
|
Provenance
Provenance describes where a file came from. On PyPI, provenance is shared via attestations, which provide a verifiable record of the build or publishing details. View details, limitations and caveats.
PyPI Publish Attestation
PyPI verified that this artifact, at this checksum, originated from the publisher listed below.
Signed by GitHub Actions, verified by PyPI on Jul 20, 2026.
Transparency log
