mDeepFRI / model_config.json
Valentyn Bezshapkin
feat: adding ONNX model weights
7aa1081
raw
history blame
818 Bytes
{
"cnn": {
"gcn": false,
"layer_name": "CNN_concatenate",
"models": {
"ec": "./onnx_deepfri_models/DeepCNN-MERGED_enzyme_commission",
"mf": "./onnx_deepfri_models/DeepCNN-MERGED_molecular_function",
"bp": "./onnx_deepfri_models/DeepCNN-MERGED_biological_process",
"cc": "./onnx_deepfri_models/DeepCNN-MERGED_cellular_component"
}
},
"gcn": {
"gcn": true,
"layer_name": "GCNN_concatenate",
"models": {
"mf": "./onnx_deepfri_models/DeepFRI-MERGED_GraphConv_gcd_512-512-512_fcd_1024_ca_10.0_mf",
"bp": "./onnx_deepfri_models/DeepFRI-MERGED_GraphConv_gcd_512-512-512_fcd_1024_ca_10.0_bp",
"cc": "./onnx_deepfri_models/DeepFRI-MERGED_GraphConv_gcd_512-512-512_fcd_1024_ca_10.0_cc",
"ec": "./onnx_deepfri_models/DeepFRI-MERGED_GraphConv_gcd_512-512-512_fcd_1024_ca_10.0_ec"
}
}
}