File size: 731 Bytes
805064d | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 | {
"model_type": "bert",
"task_specific_params": {
"text-classification": {
"model": "Rostlab/prot_bert_bfd_localization"
}
},
"id2label": {
"0": "Cell.membrane",
"1": "Cytoplasm",
"2": "Endoplasmic.reticulum",
"3": "Extracellular",
"4": "Golgi.apparatus",
"5": "Lysosome/Vacuole",
"6": "Mitochondrion",
"7": "Nucleus",
"8": "Peroxisome",
"9": "Plastid"
},
"label2id": {
"Cell.membrane": 0,
"Cytoplasm": 1,
"Endoplasmic.reticulum": 2,
"Extracellular": 3,
"Golgi.apparatus": 4,
"Lysosome/Vacuole": 5,
"Mitochondrion": 6,
"Nucleus": 7,
"Peroxisome": 8,
"Plastid": 9
},
"num_labels": 10,
"transformers_version": "4.57.1"
}
|