| { | |
| "model_type": "GIN", | |
| "in_channels": 7, | |
| "hidden_channels": 64, | |
| "num_classes": 2, | |
| "num_layers": 5, | |
| "dropout": 0.5, | |
| "aggregation": "sum", | |
| "train_eps": true, | |
| "dataset": "MUTAG", | |
| "description": "Graph Isomorphism Network trained on MUTAG dataset for graph classification. Demonstrates isomorphism-aware learning with injective SUM aggregation." | |
| } |