CASCADE-Alzheimer / config.json
valegiunca's picture
Updated files
cf36bf7 verified
Raw
History Blame Contribute Delete
1.23 kB
{
"dataset": "SEATTLE",
"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": 18370,
"max_length_up": 1024,
"max_length_down": 1024,
"temperature": 0.1,
"seed": 20,
"training_epoch": 0,
"training_global_step": 40000,
"data_composition": {
"diseases": [
"dementia",
"normal"
],
"tissues": [
"dorsolateral prefrontal cortex",
"middle temporal gyrus"
],
"cell_type_ontology_ids": [
"CL:0000128",
"CL:0000129",
"CL:0002453",
"CL:0002605",
"CL:1001602",
"CL:4023011",
"CL:4023012",
"CL:4023013",
"CL:4023015",
"CL:4023016",
"CL:4023017",
"CL:4023018",
"CL:4023036",
"CL:4023038",
"CL:4023040",
"CL:4023041",
"CL:4023051",
"CL:4023064"
]
}
}