From 52bf6f9127c465d1e1c4e1006e06f9e341d5c0eb Mon Sep 17 00:00:00 2001 From: Evie Qiu <65006489+EvieQ01@users.noreply.github.com> Date: Sun, 23 Jun 2024 20:56:15 -0400 Subject: [PATCH] Add node_names to GRaSP.py Signed-off-by: Evie Qiu <65006489+EvieQ01@users.noreply.github.com> --- causallearn/search/PermutationBased/GRaSP.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/causallearn/search/PermutationBased/GRaSP.py b/causallearn/search/PermutationBased/GRaSP.py index 38cdabf7..abcbf47c 100644 --- a/causallearn/search/PermutationBased/GRaSP.py +++ b/causallearn/search/PermutationBased/GRaSP.py @@ -81,6 +81,7 @@ def grasp( maxP: Optional[float] = None, parameters: Optional[Dict[str, Any]] = None, verbose: Optional[bool] = True, + node_names: Optional[List[str]] = None, ) -> GeneralGraph: """ Perform a greedy relaxation of the sparsest permutation (GRaSP) algorithm @@ -185,7 +186,7 @@ def grasp( raise Exception("Unknown function!") score = localScoreClass - node_names = [("x%d" % i) for i in range(p)] + node_names = [("X%d" % (i + 1)) for i in range(p)] if node_names is None else node_names nodes = [] for name in node_names: