Published November 18, 2024 | Version v1
Model Open

ScorpioBigEmbed

Authors/Creators

Description

Using ScorpioBigEmbed

Scorpio is a dynamic tool designed to enhance DNA sequence embeddings. For detailed instructions on how to load and use the model, please review ourWiki

Loading the Model


import torch
import TripletModel

# Path to the model weights
weights_p = "./checkpoint.pt"

# Load the model's state dictionary
state = torch.load(weights_p)['model_state_dict']

# Initialize the model
model = torch.load(weights_p)["Tuner"]

# Load the state dictionary into the model
model.load_state_dict(state)

Citation

If you use Scorpio in your research, please cite our work:


@article{refahi2024scorpio,
  title={Scorpio: Enhancing Embeddings to Improve Downstream Analysis of DNA sequences},
  author={Refahi, Mohammadsaleh and Sokhansanj, Bahrad A and Mell, Joshua Chang and Brown, James and Yoo, Hyunwoo and Hearne, Gavin and Rosen, Gail},
  journal={bioRxiv},
  pages={2024--07},
  year={2024},
  publisher={Cold Spring Harbor Laboratory}
}

 

Files

Files (153.7 MB)

Name Size Download all
md5:5654fb752d4b1d3b6e60a21df939c76d
153.7 MB Download
md5:5ed4ab16637cd0b0136ac1cac0e9b44d
4.2 kB Download

Additional details

Software

Repository URL
https://github.com/EESI/Scorpio
Programming language
Python