snap-stanford / UCE

UCE is a zero-shot foundation model for single-cell gene expression data
MIT License
120 stars 15 forks source link

Training UCE #25

Closed v-mahughes closed 4 months ago

v-mahughes commented 4 months ago

Hello,

Does the current UCE package/scripts provided on this page have the functionality for users to train UCE on their own reference atlas ?

Yanay1 commented 4 months ago

Since the UCE model is intended to be used as a zero shot model, we do not plan on releasing code to train the entire model.

However, you can always train downstream models on embeddings produced by one of the base (4L or 33L) models, or set up your own finetuning / training set up.