scikit-network 0.33.0__cp39-cp39-win_amd64.whl → 0.33.3__cp39-cp39-win_amd64.whl

sknetwork/clustering/metrics.py

@@ -64,7 +64,7 @@ def get_modularity(input_matrix: Union[sparse.csr_matrix, np.ndarray], labels: n
     >>> from sknetwork.data import house
     >>> adjacency = house()
     >>> labels = np.array([0, 0, 1, 1, 0])
-    >>> np.round(get_modularity(adjacency, labels), 2)
+    >>> float(np.round(get_modularity(adjacency, labels), 2))
     0.11
     """
     adjacency, bipartite = get_adjacency(input_matrix.astype(float))

sknetwork/clustering/tests/test_louvain.py

@@ -17,6 +17,12 @@ class TestLouvainClustering(unittest.TestCase):
         labels = Louvain().fit_predict(adjacency)
         self.assertEqual(len(labels), n)
 
+    def test_format(self):
+        adjacency = test_graph()
+        n = adjacency.shape[0]
+        labels = Louvain().fit_predict(adjacency.toarray())
+        self.assertEqual(len(labels), n)
+
     def test_modularity(self):
         adjacency = karate_club()
         louvain_d = Louvain(modularity='dugue')

sknetwork/gnn/base_activation.py

@@ -9,6 +9,7 @@ import numpy as np
 
 class BaseActivation:
     """Base class for activation functions.
+
     Parameters
     ----------
     name : str

sknetwork/gnn/gnn_classifier.py

@@ -93,7 +93,7 @@ class GNNClassifier(BaseGNN):
     >>> features = adjacency.copy()
     >>> gnn = GNNClassifier(dims=1, early_stopping=False)
     >>> labels_pred = gnn.fit_predict(adjacency, features, labels, random_state=42)
-    >>> round(np.mean(labels_pred == labels_true), 2)
+    >>> float(round(np.mean(labels_pred == labels_true), 2))
     0.88
     """
 

sknetwork/hierarchy/metrics.py

@@ -100,7 +100,7 @@ def dasgupta_cost(adjacency: sparse.csr_matrix, dendrogram: np.ndarray, weights:
     >>> adjacency = house()
     >>> dendrogram = paris.fit_transform(adjacency)
     >>> cost = dasgupta_cost(adjacency, dendrogram)
-    >>> np.round(cost, 2)
+    >>> float(np.round(cost, 2))
     3.33
 
     References
@@ -154,7 +154,7 @@ def dasgupta_score(adjacency: sparse.csr_matrix, dendrogram: np.ndarray, weights
     >>> adjacency = house()
     >>> dendrogram = paris.fit_transform(adjacency)
     >>> score = dasgupta_score(adjacency, dendrogram)
-    >>> np.round(score, 2)
+    >>> float(np.round(score, 2))
     0.33
 
     References
@@ -194,7 +194,7 @@ def tree_sampling_divergence(adjacency: sparse.csr_matrix, dendrogram: np.ndarra
     >>> adjacency = house()
     >>> dendrogram = paris.fit_transform(adjacency)
     >>> score = tree_sampling_divergence(adjacency, dendrogram)
-    >>> np.round(score, 2)
+    >>> float(np.round(score, 2))
     0.05
 
     References