scikit-network 0.31.0__cp310-cp310-win_amd64.whl → 0.32.1__cp310-cp310-win_amd64.whl

sknetwork/hierarchy/paris.pyx

@@ -157,21 +157,21 @@ class Paris(BaseHierarchy):
 
     Parameters
     ----------
-    weights :
+    weights : str
         Weights of nodes.
         ``'degree'`` (default) or ``'uniform'``.
-    reorder :
+    reorder : bool
         If ``True`` (default), reorder the dendrogram in non-decreasing order of height.
 
     Attributes
     ----------
-    dendrogram_ :
+    dendrogram_ : np.ndarray
         Dendrogram of the graph.
-    dendrogram_row_ :
+    dendrogram_row_ : np.ndarray
         Dendrogram for the rows, for bipartite graphs.
-    dendrogram_col_ :
+    dendrogram_col_ : np.ndarray
         Dendrogram for the columns, for bipartite graphs.
-    dendrogram_full_ :
+    dendrogram_full_ : np.ndarray
         Dendrogram for both rows and columns, indexed in this order, for bipartite graphs.
 
     Examples
@@ -217,7 +217,7 @@ class Paris(BaseHierarchy):
 
         Parameters
         ----------
-        input_matrix :
+        input_matrix : sparse.csr_matrix, np.ndarray
             Adjacency matrix or biadjacency matrix of the graph.
 
         Returns

sknetwork/hierarchy/postprocess.py

@@ -75,17 +75,17 @@ def cut_straight(dendrogram: np.ndarray, n_clusters: Optional[int] = None, thres
 
     Parameters
     ----------
-    dendrogram:
+    dendrogram : np.ndarray
         Dendrogram.
-    n_clusters :
+    n_clusters : int
         Number of clusters (optional).
         The number of clusters can be larger than n_clusters in case of equal heights in the dendrogram.
-    threshold :
+    threshold : float
         Threshold on height (optional).
         If both n_clusters and threshold are ``None``, n_clusters is set to 2.
-    sort_clusters :
+    sort_clusters : bool
         If ``True``,  sorts clusters in decreasing order of size.
-    return_dendrogram :
+    return_dendrogram : bool
         If ``True``, returns the dendrogram formed by the clusters up to the root.
     Returns
     -------
@@ -135,13 +135,13 @@ def cut_balanced(dendrogram: np.ndarray, max_cluster_size: int = 20, sort_cluste
 
     Parameters
     ----------
-    dendrogram:
+    dendrogram : np.ndarray
         Dendrogram
-    max_cluster_size :
+    max_cluster_size : int
         Maximum size of each cluster.
-    sort_clusters :
+    sort_clusters : bool
         If ``True``, sort labels in decreasing order of cluster size.
-    return_dendrogram :
+    return_dendrogram : bool
         If ``True``, returns the dendrogram formed by the clusters up to the root.
     Returns
     -------
@@ -179,19 +179,19 @@ def aggregate_dendrogram(dendrogram: np.ndarray, n_clusters: int = 2, return_cou
 
     Parameters
     ----------
-    dendrogram:
+    dendrogram : np.ndarray
         The input to aggregate.
-    n_clusters:
+    n_clusters : int
         Number of clusters (or leaves) to keep.
-    return_counts
+    return_counts :  bool
         If ``True``, returns an array of counts corresponding to the sizes of the merged subtrees.
         The sum of the counts is equal to the number of samples in the input dendrogram.
 
     Returns
     -------
-    new_dendrogram:
+    new_dendrogram : np.ndarray
         Aggregated dendrogram. The nodes are reindexed from 0.
-    counts:
+    counts : np.ndarray
         Size of the subtrees corresponding to each leaf in new_dendrogram.
     """
     n_nodes: int = dendrogram.shape[0] + 1
@@ -220,12 +220,12 @@ def get_index(tree):
 
     Parameters
     ----------
-    tree:
+    tree :
         The tree to be indexed
 
     Returns
     -------
-    index:
+    index :
         The index of the root of the given tree
     """
     if type(tree) != list:

sknetwork/hierarchy/tests/test_algos.py

@@ -27,3 +27,8 @@ class TestLouvainHierarchy(unittest.TestCase):
                 self.assertEqual(dendrogram.shape, (input_matrix.shape[0] - 1, 4))
                 if algo.bipartite:
                     self.assertEqual(algo.dendrogram_full_.shape, (sum(input_matrix.shape) - 1, 4))
+        adjacency = test_graph()
+        algo = Paris()
+        dendrogram = algo.fit_predict(adjacency)
+        dendrogram_ = algo.predict()
+        self.assertAlmostEqual(np.linalg.norm(dendrogram - dendrogram_), 0)

sknetwork/linalg/__init__.py

@@ -2,7 +2,7 @@
 from sknetwork.linalg.basics import safe_sparse_dot
 from sknetwork.linalg.eig_solver import EigSolver, LanczosEig
 from sknetwork.linalg.laplacian import get_laplacian
-from sknetwork.linalg.normalization import diagonal_pseudo_inverse, get_norms, normalize
+from sknetwork.linalg.normalizer import diagonal_pseudo_inverse, get_norms, normalize
 from sknetwork.linalg.operators import Regularizer, Laplacian, Normalizer, CoNeighbor
 from sknetwork.linalg.polynome import Polynome
 from sknetwork.linalg.sparse_lowrank import SparseLR