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