toxipredict-gnn-models / model_config.json
Arko007's picture
Upload model_config.json with huggingface_hub
81b3424 verified
Raw
History Blame Contribute Delete
417 Bytes
{
"node_dim": 45,
"edge_dim": 11,
"hidden_dim": 128,
"num_tasks": 10,
"dropout": 0.15,
"task_names": [
"NR-AR",
"NR-AhR",
"NR-Aromatase",
"NR-ER",
"NR-PPAR-gamma",
"SR-ARE",
"SR-ATAD5",
"SR-HSE",
"SR-MMP",
"SR-p53"
],
"architecture": "MultiTaskGNN_ResGATv2_JK_VN",
"version": "1.3.0",
"cv_auc_mean": 0.7855697746988419,
"cv_auc_std": 0.03941078105764019
}