Downloads · 30 days
18
41% of all-time downloads
Moonn1205/AntigenLM-subtype-classifier
AntigenLM-subtype-classifier is a machine learning model from Moonn1205. Use it for the machine learning task on the model card, and read the license before you ship it in a product. It is set up for transformers. The card lists the license as mit.
Subtype classification head of AntigenLM. Given an <HA...<NA...<sep input, predicts one of 12 influenza A subtypes (H1N1, H3N2, H5N1, ...).
Downloads · 30 days
18
41% of all-time downloads
All-time downloads
44
Public
Repo size
1.1 GB
Likes
0
Public
Click a slice to open those files.
.bin1.1 GB · 100%
From the Hugging Face model README
Subtype classification head of AntigenLM. Given an <HA>...<NA>...<sep> input, predicts one of 12 influenza A subtypes (H1N1, H3N2, H5N1, ...).
Paper: AntigenLM: Structure-Aware DNA Language Modeling for Influenza (arXiv:2602.09067). Code: https://github.com/Moonn1205/AntigenLM <!-- update with the real repo URL -->
pip install huggingface_hub transformers==4.29.2 torch==1.13.1
from huggingface_hub import snapshot_download
snapshot_download(
repo_id="Moonn1205/AntigenLM-subtype-classifier",
local_dir="subtype_classifier",
local_dir_use_symlinks=False,
)
Then follow the instructions in the code repository to run inference.
pytorch_model.bin - model weightsconfig.json - GPT-2 backbone config + (classifier only) label2id / id2labelvocab.json, tokenizer_config.json, special_tokens_map.json - tokenizer