From 0b6f6e3a8d34497c02389b2fe7cfbeb967e399e1 Mon Sep 17 00:00:00 2001 From: michellemli Date: Mon, 23 Sep 2024 23:26:51 -0400 Subject: [PATCH 1/2] Add node names instead of node IDs to networkx graph --- tdc/resource/primekg.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tdc/resource/primekg.py b/tdc/resource/primekg.py index ae0df10a..8a520ca0 100644 --- a/tdc/resource/primekg.py +++ b/tdc/resource/primekg.py @@ -33,8 +33,8 @@ def to_nx(self): G = nx.Graph() for i in self.df.relation.unique(): - G.add_edges_from(self.df[self.df.relation == i][["x_id", - "y_id"]].values, + G.add_edges_from(self.df[self.df.relation == i][["x_name", + "y_name"]].values, relation=i) return G From bf728294c350cbcd213f8056baaf387dc93c407e Mon Sep 17 00:00:00 2001 From: michellemli Date: Tue, 24 Sep 2024 00:05:14 -0400 Subject: [PATCH 2/2] Fix formatting for yapf --- tdc/resource/primekg.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/tdc/resource/primekg.py b/tdc/resource/primekg.py index 8a520ca0..38935b24 100644 --- a/tdc/resource/primekg.py +++ b/tdc/resource/primekg.py @@ -33,9 +33,9 @@ def to_nx(self): G = nx.Graph() for i in self.df.relation.unique(): - G.add_edges_from(self.df[self.df.relation == i][["x_name", - "y_name"]].values, - relation=i) + G.add_edges_from( + self.df[self.df.relation == i][["x_name", "y_name"]].values, + relation=i) return G def get_features(self, feature_type):