tahoe-lung-classifier

Fine-tuned Tahoe-X1 model for cell type classification on lung tissue data.

Model Details

• Base Model: Tahoe-X1 (70M parameters)
• Fine-tuning Method: LoRA (r=8, alpha=16)
• Task: Cell type classification
• Classes: 51 cell types
• Dataset: Lung tissue single-cell RNA-seq

Usage

from inference import load_model, predict_cell_types

# Load model
model, vocab, collator_cfg = load_model(
    model_path=".",
    device="cuda"
)

# Predict on h5ad file
predictions = predict_cell_types(
    model=model,
    vocab=vocab,
    collator_cfg=collator_cfg,
    h5ad_path="your_data.h5ad",
    label_key="cell_type",  # Optional: for evaluation
)

Cell Types

• B cell
• CD1c-positive myeloid dendritic cell
• CD4-positive, alpha-beta T cell
• CD8-positive, alpha-beta T cell
• T cell
• acinar cell
• alveolar adventitial fibroblast
• alveolar macrophage
• alveolar type 1 fibroblast cell
• bronchial goblet cell
• bronchus fibroblast of lung
• brush cell of tracheobronchial tree
• capillary endothelial cell
• classical monocyte
• club cell
• conventional dendritic cell
• dendritic cell
• elicited macrophage
• endothelial cell of lymphatic vessel
• epithelial cell of lower respiratory tract
• fibroblast
• hematopoietic stem cell
• ionocyte
• lung macrophage
• lung pericyte
• mast cell
• mesothelial cell
• mucus secreting cell
• multiciliated columnar cell of tracheobronchial tree
• multiciliated epithelial cell
• myofibroblast cell
• nasal mucosa goblet cell
• natural killer cell
• non-classical monocyte
• plasma cell
• plasmacytoid dendritic cell
• pulmonary alveolar epithelial cell
• pulmonary alveolar type 1 cell
• pulmonary alveolar type 2 cell
• pulmonary artery endothelial cell
• pulmonary neuroendocrine cell
• respiratory basal cell
• respiratory tract hillock cell
• serous secreting cell
• smooth muscle cell
• stromal cell
• tracheobronchial goblet cell
• tracheobronchial serous cell
• tracheobronchial smooth muscle cell
• unknown
• vein endothelial cell

Citation

If you use this model, please cite:

• Tahoe-X1: https://huggingface.co/tahoebio/Tahoe-x1