tahoe-lung-classifier / config.json
Yuto2007's picture
Upload folder using huggingface_hub
3147616 verified
{
"model_type": "tahoe-cell-classifier",
"architectures": [
"CellTypeClassifier"
],
"base_model": "tahoebio/Tahoe-x1",
"model_size": "70m",
"num_labels": 51,
"d_model": 512,
"n_layers": 12,
"n_heads": 8,
"expansion_ratio": 4,
"classifier_dropout": 0.1,
"label2id": {
"B cell": 0,
"CD1c-positive myeloid dendritic cell": 1,
"CD4-positive, alpha-beta T cell": 2,
"CD8-positive, alpha-beta T cell": 3,
"T cell": 4,
"acinar cell": 5,
"alveolar adventitial fibroblast": 6,
"alveolar macrophage": 7,
"alveolar type 1 fibroblast cell": 8,
"bronchial goblet cell": 9,
"bronchus fibroblast of lung": 10,
"brush cell of tracheobronchial tree": 11,
"capillary endothelial cell": 12,
"classical monocyte": 13,
"club cell": 14,
"conventional dendritic cell": 15,
"dendritic cell": 16,
"elicited macrophage": 17,
"endothelial cell of lymphatic vessel": 18,
"epithelial cell of lower respiratory tract": 19,
"fibroblast": 20,
"hematopoietic stem cell": 21,
"ionocyte": 22,
"lung macrophage": 23,
"lung pericyte": 24,
"mast cell": 25,
"mesothelial cell": 26,
"mucus secreting cell": 27,
"multiciliated columnar cell of tracheobronchial tree": 28,
"multiciliated epithelial cell": 29,
"myofibroblast cell": 30,
"nasal mucosa goblet cell": 31,
"natural killer cell": 32,
"non-classical monocyte": 33,
"plasma cell": 34,
"plasmacytoid dendritic cell": 35,
"pulmonary alveolar epithelial cell": 36,
"pulmonary alveolar type 1 cell": 37,
"pulmonary alveolar type 2 cell": 38,
"pulmonary artery endothelial cell": 39,
"pulmonary neuroendocrine cell": 40,
"respiratory basal cell": 41,
"respiratory tract hillock cell": 42,
"serous secreting cell": 43,
"smooth muscle cell": 44,
"stromal cell": 45,
"tracheobronchial goblet cell": 46,
"tracheobronchial serous cell": 47,
"tracheobronchial smooth muscle cell": 48,
"unknown": 49,
"vein endothelial cell": 50
},
"id2label": {
"0": "B cell",
"1": "CD1c-positive myeloid dendritic cell",
"2": "CD4-positive, alpha-beta T cell",
"3": "CD8-positive, alpha-beta T cell",
"4": "T cell",
"5": "acinar cell",
"6": "alveolar adventitial fibroblast",
"7": "alveolar macrophage",
"8": "alveolar type 1 fibroblast cell",
"9": "bronchial goblet cell",
"10": "bronchus fibroblast of lung",
"11": "brush cell of tracheobronchial tree",
"12": "capillary endothelial cell",
"13": "classical monocyte",
"14": "club cell",
"15": "conventional dendritic cell",
"16": "dendritic cell",
"17": "elicited macrophage",
"18": "endothelial cell of lymphatic vessel",
"19": "epithelial cell of lower respiratory tract",
"20": "fibroblast",
"21": "hematopoietic stem cell",
"22": "ionocyte",
"23": "lung macrophage",
"24": "lung pericyte",
"25": "mast cell",
"26": "mesothelial cell",
"27": "mucus secreting cell",
"28": "multiciliated columnar cell of tracheobronchial tree",
"29": "multiciliated epithelial cell",
"30": "myofibroblast cell",
"31": "nasal mucosa goblet cell",
"32": "natural killer cell",
"33": "non-classical monocyte",
"34": "plasma cell",
"35": "plasmacytoid dendritic cell",
"36": "pulmonary alveolar epithelial cell",
"37": "pulmonary alveolar type 1 cell",
"38": "pulmonary alveolar type 2 cell",
"39": "pulmonary artery endothelial cell",
"40": "pulmonary neuroendocrine cell",
"41": "respiratory basal cell",
"42": "respiratory tract hillock cell",
"43": "serous secreting cell",
"44": "smooth muscle cell",
"45": "stromal cell",
"46": "tracheobronchial goblet cell",
"47": "tracheobronchial serous cell",
"48": "tracheobronchial smooth muscle cell",
"49": "unknown",
"50": "vein endothelial cell"
},
"task": "cell_type_classification"
}