You need to agree to share your contact information to access this model

This repository is publicly accessible, but you have to accept the conditions to access its files and content.

Log in or Sign Up to review the conditions and access this model content.

vHAMSTeR Models v1.3.0

Model weights for vHAMSTeR (Virus Host Assignment Model using Sequence Transformers and Reading-frames), a tool for predicting the host of a virus from its genomic sequence.

Model description

vHAMSTeR predicts viral host as one of five classes: 'Prokaryote', 'Animal', 'Plant', 'Fungi' or 'Protists, Algae & Oomycetes'.

It is designed for viral sequences up to 10 kbp. Longer sequences are split into 10 kbp chunks, scored independently, and aggregated to produce a genome-level consensus.

Architecture

Each fold in the ensemble consists of:

  1. PEFT adapter on InstaDeepAI/NTv3_650M_pre β€” a 650M-parameter nucleotide language model fine-tuned on viral sequences
  2. XGBoost stacking layer β€” incorporates geNomad marker features alongside the transformer embeddings
  3. GLM gate β€” learned weighting between the transformer and marker feature streams
  4. Classifier head β€” produces the final 5-class probability distribution

Predictions are calibrated using length-aware vector scaling (length_aware_vector_scaling_anchors_5.json).

The full ensemble runs 5 cross-validation folds and aggregates their outputs.

Repository structure

  vhamster_models_v1.3.0/
β”œβ”€β”€ length_aware_vector_scaling_anchors_5.json   # calibration parameters
β”œβ”€β”€ fold_0/
β”‚   β”œβ”€β”€ config.json
β”‚   β”œβ”€β”€ best_macro_auprc_model/
β”‚   β”‚   β”œβ”€β”€ adapter_config.json
β”‚   β”‚   β”œβ”€β”€ adapter_model.safetensors
β”‚   β”‚   β”œβ”€β”€ classifier_head.pt
β”‚   β”‚   β”œβ”€β”€ tokenizer_config.json
β”‚   β”‚   β”œβ”€β”€ tokenization_ntv3.py
β”‚   β”‚   └── vocab.json
β”‚   └── xgb_stacking_artifacts/
β”‚       β”œβ”€β”€ xgb_stacking_artifacts.json
β”‚       β”œβ”€β”€ xgb1_binary_model.json
β”‚       β”œβ”€β”€ xgb2_euk_model.json
β”‚       β”œβ”€β”€ marker_classification.pkl
β”‚       └── marker_cutoffs.pkl
β”œβ”€β”€ fold_1/ … fold_4/   # same structure

Usage

These weights are intended to be used via the vHAMSTeR command-line tool. See the vHAMSTeR repository for full installation instructions.

Downloads last month

-

Downloads are not tracked for this model. How to track
Inference Providers NEW
This model isn't deployed by any Inference Provider. πŸ™‹ Ask for provider support

Model tree for DOEJGI/vhamster-models

Finetuned
(2)
this model