---
tags:
- single-cell
- cell-type-classification
- biology
- tahoe-x1
library_name: transformers
---

# 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

```python
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