Downloads · 30 days
14
13% of all-time downloads
genbio-ai/GB.RNA-1.6B-mrna-expression-level-pc3
GB.RNA-1.6B-mrna-expression-level-pc3 is a machine learning model from genbio-ai. 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 other.
10-fold cross-validation fully fine-tuned checkpoints for mRNA expression level prediction (pc3).
Downloads · 30 days
14
13% of all-time downloads
All-time downloads
107
Public
Repo size
386 GB
Likes
0
Public
Click a slice to open those files.
.ckpt193 GB · 100%
From the Hugging Face model README
10-fold cross-validation fully fine-tuned checkpoints for mRNA expression level prediction (pc3).
from huggingface_hub import snapshot_download
from pathlib import Path
model_name = "genbio-ai/GB.RNA-1.6B-mrna-expression-level-pc3"
genbio_models_path = Path.home().joinpath('genbio_models', model_name)
genbio_models_path.mkdir(parents=True, exist_ok=True)
snapshot_download(repo_id=model_name, local_dir=genbio_models_path)
from modelgenerator.tasks import SequenceRegression
ckpt_path = genbio_models_path.joinpath('fold0', 'model.ckpt')
model = SequenceRegression.load_from_checkpoint(ckpt_path, strict_loading=False).eval()
collated_batch = model.transform({"sequences": ["ACGT", "AGCT"]})
logits = model(collated_batch)
print(logits)