Downloads · 30 days
1
5% of all-time downloads
viral-sequence-prediction
nleroy917/viral-sequence-prediction
viral-sequence-prediction is a machine learning model from nleroy917. Use it for the machine learning task on the model card, and read the license before you ship it in a product.
This model was a part of the University of Virginia's NLP course final project: NLP for social good. Using publically available data from the NCBI virus data portal, we designed an LSTM-based classifier that can predi…
All-time downloads
20
Public
Repo size
636 KB
Likes
1
Public
Repo makeup
Click a slice to open those files.
.pt636 KB · 100%
At a glance
- Access
- Public
- Created
- Dec 14, 2023
- Updated
- Dec 14, 2023
- SHA
- fdbffc3a
Model card
From the Hugging Face model README
Model Card for viral-sequence-prediction
This model was a part of the University of Virginia's NLP course final project: NLP for social good. Using publically available data from the NCBI virus data portal, we designed an LSTM-based classifier that can predict the virus of origin for any arbitrary protein sequence.
Specifically, we were interested in classifying COVID v Flu.
Using the model
Code for the model can be found at: https://github.com/nleroy917/CS6501-final
You can use the train.ipynb notebook, or use the code directly:
from dna_classification.models import DNASequenceClassifier
model = DNASequenceClassifier("nleroy917/viral-sequence-prediction")
virus = model.predict("MGYINVFAFPFTIYSLLLCRMNFRNYIAQVDVVNFNLT")
>>> COVID
- Created
- Dec 14, 2023
- Updated
- Dec 14, 2023