Force type of matrix indices in Louvain

sknetwork-team · Jun 7, 2024 · ec8d3ca · ec8d3ca
1 parent 27aa364
commit ec8d3ca
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/sknetwork/clustering/louvain.py b/sknetwork/clustering/louvain.py
@@ -131,9 +131,9 @@ def _optimize(self, labels, adjacency, out_weights, in_weights):
         increase :
             Gain in modularity after optimization.
         """
-        labels = labels.astype(np.int32)
-        indices = adjacency.indices
-        indptr = adjacency.indptr
+        labels = labels.astype(np.int64)
+        indices = adjacency.indices.astype(np.int64)
+        indptr = adjacency.indptr.astype(np.int64)
         data = adjacency.data.astype(np.float32)
         out_weights = out_weights.astype(np.float32)
         in_weights = in_weights.astype(np.float32)