Downloads · 30 days
0
bdpedigo/synapse_target_hks_rf
synapse_target_hks_rf is a machine learning model from bdpedigo. Use it for the machine learning task on the model card, and read the license before you ship it in a product. The card lists the license as mit.
Random forest models trained on heat kernel signature (HKS) features to predict the compartment target of a synapse on neuronal meshes. Method is described in more detail in Pedigo et al. bioRxiv 2026.
Downloads · 30 days
0
Access
Public
Updated Feb 23, 2026
Repo size
279 MB
Likes
0
Public
Click a slice to open those files.
.joblib279 MB · 100%
From the Hugging Face model README
Random forest models trained on heat kernel signature (HKS) features to predict the compartment target of a synapse on neuronal meshes. Method is described in more detail in Pedigo et al. bioRxiv 2026.
The feature generation pipeline used here can be found in the meshmash repo.
The parameters for that function are in parameters.toml.
synapse_hks_model.joblib takes in the 32 HKS features generated by that pipeline, as well as a column of the distance from each point to the postsynaptic nucleus.
The model outputs predictions for {"spine", "shaft", "soma"} targets. This is the model used in the paper.
simple_hks_model.joblib uses only the HKS features and outputs {"spine", "not_spine"} labels.
Please see the paper above for much more detail on the performance of the classifier and its application. Questions and comments are more than welcome: [email protected].