This Github repository contains all models implemented for our CS271 Final Project: Predicting Cell Type from Spatial Transcriptomic Data. In total, we have notebooks for 6 models:
- A Baseline MLP model
- A Periphery Score MLP model
- A Cell Boundary CNN
- A ResNet Cell Boundary Model
- A Late Fusion Multimodal Model that linearly combines models 2 and 3
- An early-fusion Transformer model
We used the MERFISH dataset, specially mouse1sample1.hdf5. Periphery scores were calculated and given to us by our external mentors at the Salzman lab.
Code written by: Ayushi Tandel and Nathaniel Chien