Fix homophily order

CSHVienna · Sep 11, 2024 · c4a8feb · c4a8feb
1 parent 6d91d6c
commit c4a8feb
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/netin/models/dh_model.py b/netin/models/dh_model.py
@@ -54,7 +54,7 @@ def _initialize_lfms(self):
         super()._initialize_lfms()
         self.h = TwoClassHomophily.from_two_class_homophily(
             node_class_values=self.graph.get_node_class(CLASS_ATTRIBUTE),
-            homophily=(self.h_m, self.h_M))
+            homophily=(self.h_M, self.h_m))
 
     def compute_target_probabilities(self, source: int)\
         -> np.ndarray:

diff --git a/netin/models/dpah_model.py b/netin/models/dpah_model.py
@@ -58,7 +58,7 @@ def _initialize_lfms(self):
         super()._initialize_lfms()
         self.h = TwoClassHomophily.from_two_class_homophily(
             node_class_values=self.graph.get_node_class(CLASS_ATTRIBUTE),
-            homophily=(self.h_m, self.h_M))
+            homophily=(self.h_M, self.h_m))
         self.pa = InDegreePreferentialAttachment(
             graph=self.graph, N=self._n_nodes_total)
 

diff --git a/netin/models/homophily_model.py b/netin/models/homophily_model.py
@@ -58,4 +58,4 @@ def compute_target_probabilities(self, source: int) -> np.ndarray:
     def _initialize_lfms(self):
         self.h = TwoClassHomophily.from_two_class_homophily(
             node_class_values=self.graph.get_node_class(CLASS_ATTRIBUTE),
-            homophily=(self.h_m, self.h_M))
+            homophily=(self.h_M, self.h_m))