{ "nfeature_dim": 6, "efeature_dim": 5, "output_dim": 6, "hidden_dim": 512, "n_gnn_layers": 5, "K": 3, "dropout_rate": 0.2 }