CASCADE-LUCA / config.json
valegiunca's picture
Updated files
177024c verified
Raw
History Blame Contribute Delete
1.7 kB
{
"dataset": "LUCA",
"architecture": "TransformerGenerator",
"d_model": 384,
"nhead": 6,
"nlayers": 12,
"dim_embedding": 384,
"dropout": 0.1,
"nclass": 2,
"pad_token": "<pad>",
"cell_emb_style": "avg-pool",
"context_specific_projections": true,
"constant_ctx": false,
"only_contrastive": true,
"DA": true,
"lambda_sinkhorn": 0.005,
"contexts": [
"disease",
"cell_type",
"tissue"
],
"merged_contexts": "disease_cell_type_tissue",
"vocab_size": 14847,
"max_length_up": 1024,
"max_length_down": 1024,
"temperature": 0.1,
"seed": 20,
"training_epoch": 1,
"training_global_step": 400000,
"data_composition": {
"diseases": [
"chronic obstructive pulmonary disease",
"lung adenocarcinoma",
"non-small cell lung carcinoma",
"normal",
"squamous cell lung carcinoma"
],
"tissues": [
"adrenal tissue",
"brain",
"liver",
"lung",
"lymph node",
"pleural effusion"
],
"cell_type_ontology_ids": [
"CL:0000077",
"CL:0000082",
"CL:0000097",
"CL:0000158",
"CL:0000192",
"CL:0000235",
"CL:0000236",
"CL:0000451",
"CL:0000499",
"CL:0000583",
"CL:0000623",
"CL:0000624",
"CL:0000625",
"CL:0000669",
"CL:0000763",
"CL:0000775",
"CL:0000784",
"CL:0000786",
"CL:0000815",
"CL:0000860",
"CL:0000875",
"CL:0000990",
"CL:0001064",
"CL:0002062",
"CL:0002063",
"CL:0002138",
"CL:0002144",
"CL:0002399",
"CL:0002543",
"CL:0002553",
"CL:0005012",
"CL:1001568",
"CL:2000093"
]
}
}