Genentech/human-atac-catlas-model

human-chromhmm-fullstack-model

Model Description

This model is a multi-class classifier trained to classify genomic DNA sequences into one of 16 classes. It was trained by fine-tuning the Enformer model using the grelu library on top of the human ChromHMM fullstack annotation dataset.

• Architecture: Fine-tuned Enformer
• Input: Genomic sequences (hg38)
• Output: Probability across 16 class labels.

Repository Content

1. model.ckpt: The trained model weights and hyperparameters (PyTorch Lightning checkpoint).
2. 2_train.ipynb: Jupyter notebook containing the training logic, architecture definition, and evaluation loops.
3. output.log: Training logs.

How to use

To load this model for inference or fine-tuning, use the grelu interface:

from grelu.lightning import LightningModel
from huggingface_hub import hf_hub_download

ckpt_path = hf_hub_download(
    repo_id="Genentech/human-chromhmm-fullstack-model", 
    filename="model.ckpt"
)

model = LightningModel.load_from_checkpoint(ckpt_path)
model.eval()