Instructions to use ctheodoris/Geneformer with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use ctheodoris/Geneformer with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("fill-mask", model="ctheodoris/Geneformer")# Load model directly from transformers import AutoTokenizer, AutoModelForMaskedLM tokenizer = AutoTokenizer.from_pretrained("ctheodoris/Geneformer") model = AutoModelForMaskedLM.from_pretrained("ctheodoris/Geneformer") - Inference
- Notebooks
- Google Colab
- Kaggle
Clarification on label mapping for cardiomyopathy classifier (nf, dcm, hcm)
#571
by ro2h5 - opened
Hi,
I’m using the fine-tuned cardiomyopathy cell classifier from your repository:
https://huggingface.co/ctheodoris/Geneformer/blob/main/fine_tuned_models/Geneformer-V1-10M_CellClassifier_cardiomyopathies_220224/config.json
The config.json shows:
id2label": {
"0": "LABEL_0",
"1": "LABEL_1",
"2": "LABEL_2"
},
"label2id": {
"LABEL_0": 0,
"LABEL_1": 1,
"LABEL_2": 2
}
Could you please clarify how these labels map to the disease states: nf, dcm, and hcm?
Thanks!
Thank you for your question: 0: “nf”, 1: “hcm”, 2: “dcm”
ctheodoris changed discussion status to closed