scikit-network 0.31.0__cp39-cp39-win_amd64.whl → 0.32.1__cp39-cp39-win_amd64.whl

sknetwork/utils/format.py

@@ -30,7 +30,7 @@ def directed2undirected(adjacency: Union[sparse.csr_matrix, SparseLR],
 
     or
 
-    :math:`\\max(A,A^T)`
+    :math:`\\max(A,A^T) > 0` (binary)
 
     If the initial adjacency matrix :math:`A` is binary, bidirectional edges have weight 2
     (first method, default) or 1 (second method).

sknetwork/utils/membership.py

@@ -20,9 +20,9 @@ def get_membership(labels: np.ndarray, dtype=bool, n_labels: Optional[int] = Non
     Parameters
     ----------
     labels :
-        Label of each node.
+        Label of each node (integers).
     dtype :
-        Type of the entries. Boolean by default.
+        Type of the output. Boolean by default.
     n_labels : int
         Number of labels.
 

sknetwork/visualization/__init__.py

@@ -1,4 +1,4 @@
 """Visualization module."""
 
-from sknetwork.visualization.dendrograms import svg_dendrogram
-from sknetwork.visualization.graphs import svg_graph, svg_bigraph
+from sknetwork.visualization.dendrograms import visualize_dendrogram, svg_dendrogram
+from sknetwork.visualization.graphs import visualize_graph, visualize_bigraph, svg_graph, svg_bigraph

sknetwork/visualization/dendrograms.py

@@ -1,7 +1,7 @@
 #!/usr/bin/env python3
 # -*- coding: utf-8 -*-
 """
-Created on April 2020
+Created in April 2020
 @author: Thomas Bonald <bonald@enst.fr>
 """
 from typing import Iterable, Optional
@@ -158,12 +158,12 @@ def svg_dendrogram_left(dendrogram, names, width, height, margin, margin_text, s
     return svg
 
 
-def svg_dendrogram(dendrogram: np.ndarray, names: Optional[np.ndarray] = None, rotate: bool = False, width: float = 400,
-                   height: float = 300, margin: float = 10, margin_text: float = 5, scale: float = 1,
-                   line_width: float = 2, n_clusters: int = 2, color: str = 'black', colors: Optional[Iterable] = None,
-                   font_size: int = 12, reorder: bool = False, rotate_names: bool = True,
-                   filename: Optional[str] = None):
-    """Return SVG image of a dendrogram.
+def visualize_dendrogram(dendrogram: np.ndarray, names: Optional[np.ndarray] = None, rotate: bool = False,
+                         width: float = 400, height: float = 300, margin: float = 10, margin_text: float = 5,
+                         scale: float = 1, line_width: float = 2, n_clusters: int = 2, color: str = 'black',
+                         colors: Optional[Iterable] = None, font_size: int = 12, reorder: bool = False,
+                         rotate_names: bool = True, filename: Optional[str] = None):
+    """Return the image of a dendrogram in SVG format.
 
     Parameters
     ----------
@@ -227,3 +227,51 @@ def svg_dendrogram(dendrogram: np.ndarray, names: Optional[np.ndarray] = None, r
             f.write(svg)
 
     return svg
+
+
+def svg_dendrogram(dendrogram: np.ndarray, names: Optional[np.ndarray] = None, rotate: bool = False, width: float = 400,
+                   height: float = 300, margin: float = 10, margin_text: float = 5, scale: float = 1,
+                   line_width: float = 2, n_clusters: int = 2, color: str = 'black', colors: Optional[Iterable] = None,
+                   font_size: int = 12, reorder: bool = False, rotate_names: bool = True,
+                   filename: Optional[str] = None):
+    """Return the image of a dendrogram in SVG format.
+
+    Alias for visualize_dendrogram.
+
+    Parameters
+    ----------
+    dendrogram :
+        Dendrogram to display.
+    names :
+        Names of leaves.
+    rotate :
+        If ``True``, rotate the tree so that the root is on the left.
+    width :
+        Width of the image (margins excluded).
+    height :
+        Height of the image (margins excluded).
+    margin :
+        Margin.
+    margin_text :
+        Margin between leaves and their names, if any.
+    scale :
+        Scaling factor.
+    line_width :
+        Line width.
+    n_clusters :
+        Number of coloured clusters to display.
+    color :
+        Default SVG color for the dendrogram.
+    colors :
+        SVG colors of the clusters of the dendrogram (optional).
+    font_size :
+        Font size.
+    reorder :
+        If ``True``, reorder leaves so that left subtree has more leaves than right subtree.
+    rotate_names :
+        If ``True``, rotate names of leaves (only valid if **rotate** is ``False``).
+    filename :
+        Filename for saving image (optional).
+    """
+    return visualize_dendrogram(dendrogram, names, rotate, width, height, margin, margin_text, scale, line_width,
+                                n_clusters, color, colors, font_size, reorder, rotate_names, filename)

sknetwork/visualization/graphs.py

@@ -2,9 +2,8 @@
 # -*- coding: utf-8 -*-
 """
 Created in April 2020
-@authors:
-Thomas Bonald <thomas.bonald@telecom-paris.fr>
-Quentin Lutz <qlutz@live.fr>
+@author: Thomas Bonald <thomas.bonald@telecom-paris.fr>
+@author: Quentin Lutz <qlutz@live.fr>
 """
 from typing import Optional, Iterable, Union, Tuple
 
@@ -357,21 +356,22 @@ def svg_text(pos, text, margin_text, font_size=12, position: str = 'right'):
     return """<text text-anchor="{}" x="{}" y="{}" font-size="{}">{}</text>""".format(anchor, x, y, font_size, text)
 
 
-def svg_graph(adjacency: Optional[sparse.csr_matrix] = None, position: Optional[np.ndarray] = None,
-              names: Optional[np.ndarray] = None, labels: Optional[Iterable] = None,
-              name_position: str = 'right', scores: Optional[Iterable] = None,
-              probs: Optional[Union[np.ndarray, sparse.csr_matrix]] = None,
-              seeds: Union[list, dict] = None, width: Optional[float] = 400, height: Optional[float] = 300,
-              margin: float = 20, margin_text: float = 3, scale: float = 1, node_order: Optional[np.ndarray] = None,
-              node_size: float = 7, node_size_min: float = 1, node_size_max: float = 20,
-              display_node_weight: Optional[bool] = None, node_weights: Optional[np.ndarray] = None,
-              node_width: float = 1, node_width_max: float = 3, node_color: str = 'gray',
-              display_edges: bool = True, edge_labels: Optional[list] = None,
-              edge_width: float = 1, edge_width_min: float = 0.5,
-              edge_width_max: float = 20, display_edge_weight: bool = True,
-              edge_color: Optional[str] = None, label_colors: Optional[Iterable] = None,
-              font_size: int = 12, directed: Optional[bool] = None, filename: Optional[str] = None) -> str:
-    """Return SVG image of a graph.
+def visualize_graph(adjacency: Optional[sparse.csr_matrix] = None, position: Optional[np.ndarray] = None,
+                    names: Optional[np.ndarray] = None, labels: Optional[Iterable] = None,
+                    name_position: str = 'right', scores: Optional[Iterable] = None,
+                    probs: Optional[Union[np.ndarray, sparse.csr_matrix]] = None,
+                    seeds: Union[list, dict] = None, width: Optional[float] = 400, height: Optional[float] = 300,
+                    margin: float = 20, margin_text: float = 3, scale: float = 1,
+                    node_order: Optional[np.ndarray] = None, node_size: float = 7, node_size_min: float = 1,
+                    node_size_max: float = 20,
+                    display_node_weight: Optional[bool] = None, node_weights: Optional[np.ndarray] = None,
+                    node_width: float = 1, node_width_max: float = 3, node_color: str = 'gray',
+                    display_edges: bool = True, edge_labels: Optional[list] = None,
+                    edge_width: float = 1, edge_width_min: float = 0.5,
+                    edge_width_max: float = 20, display_edge_weight: bool = True,
+                    edge_color: Optional[str] = None, label_colors: Optional[Iterable] = None,
+                    font_size: int = 12, directed: Optional[bool] = None, filename: Optional[str] = None) -> str:
+    """Return the image of a graph in SVG format.
 
     Parameters
     ----------
@@ -453,8 +453,8 @@ def svg_graph(adjacency: Optional[sparse.csr_matrix] = None, position: Optional[
     >>> graph = karate_club(True)
     >>> adjacency = graph.adjacency
     >>> position = graph.position
-    >>> from sknetwork.visualization import svg_graph
-    >>> image = svg_graph(adjacency, position)
+    >>> from sknetwork.visualization import visualize_graph
+    >>> image = visualize_graph(adjacency, position)
     >>> image[1:4]
     'svg'
     """
@@ -569,28 +569,28 @@ def svg_graph(adjacency: Optional[sparse.csr_matrix] = None, position: Optional[
     return svg
 
 
-def svg_bigraph(biadjacency: sparse.csr_matrix,
-                names_row: Optional[np.ndarray] = None, names_col: Optional[np.ndarray] = None,
-                labels_row: Optional[Union[dict, np.ndarray]] = None,
-                labels_col: Optional[Union[dict, np.ndarray]] = None,
-                scores_row: Optional[Union[dict, np.ndarray]] = None,
-                scores_col: Optional[Union[dict, np.ndarray]] = None,
-                probs_row: Optional[Union[np.ndarray, sparse.csr_matrix]] = None,
-                probs_col: Optional[Union[np.ndarray, sparse.csr_matrix]] = None,
-                seeds_row: Union[list, dict] = None, seeds_col: Union[list, dict] = None,
-                position_row: Optional[np.ndarray] = None, position_col: Optional[np.ndarray] = None,
-                reorder: bool = True, width: Optional[float] = 400,
-                height: Optional[float] = 300, margin: float = 20, margin_text: float = 3, scale: float = 1,
-                node_size: float = 7, node_size_min: float = 1, node_size_max: float = 20,
-                display_node_weight: bool = False,
-                node_weights_row: Optional[np.ndarray] = None, node_weights_col: Optional[np.ndarray] = None,
-                node_width: float = 1, node_width_max: float = 3,
-                color_row: str = 'gray', color_col: str = 'gray', label_colors: Optional[Iterable] = None,
-                display_edges: bool = True, edge_labels: Optional[list] = None, edge_width: float = 1,
-                edge_width_min: float = 0.5, edge_width_max: float = 10, edge_color: str = 'black',
-                display_edge_weight: bool = True,
-                font_size: int = 12, filename: Optional[str] = None) -> str:
-    """Return SVG image of a bigraph.
+def visualize_bigraph(biadjacency: sparse.csr_matrix,
+                      names_row: Optional[np.ndarray] = None, names_col: Optional[np.ndarray] = None,
+                      labels_row: Optional[Union[dict, np.ndarray]] = None,
+                      labels_col: Optional[Union[dict, np.ndarray]] = None,
+                      scores_row: Optional[Union[dict, np.ndarray]] = None,
+                      scores_col: Optional[Union[dict, np.ndarray]] = None,
+                      probs_row: Optional[Union[np.ndarray, sparse.csr_matrix]] = None,
+                      probs_col: Optional[Union[np.ndarray, sparse.csr_matrix]] = None,
+                      seeds_row: Union[list, dict] = None, seeds_col: Union[list, dict] = None,
+                      position_row: Optional[np.ndarray] = None, position_col: Optional[np.ndarray] = None,
+                      reorder: bool = True, width: Optional[float] = 400,
+                      height: Optional[float] = 300, margin: float = 20, margin_text: float = 3, scale: float = 1,
+                      node_size: float = 7, node_size_min: float = 1, node_size_max: float = 20,
+                      display_node_weight: bool = False,
+                      node_weights_row: Optional[np.ndarray] = None, node_weights_col: Optional[np.ndarray] = None,
+                      node_width: float = 1, node_width_max: float = 3,
+                      color_row: str = 'gray', color_col: str = 'gray', label_colors: Optional[Iterable] = None,
+                      display_edges: bool = True, edge_labels: Optional[list] = None, edge_width: float = 1,
+                      edge_width_min: float = 0.5, edge_width_max: float = 10, edge_color: str = 'black',
+                      display_edge_weight: bool = True,
+                      font_size: int = 12, filename: Optional[str] = None) -> str:
+    """Return the image of a bipartite graph in SVG format.
 
     Parameters
     ----------
@@ -682,8 +682,8 @@ def svg_bigraph(biadjacency: sparse.csr_matrix,
     -------
     >>> from sknetwork.data import movie_actor
     >>> biadjacency = movie_actor()
-    >>> from sknetwork.visualization import svg_bigraph
-    >>> image = svg_bigraph(biadjacency)
+    >>> from sknetwork.visualization import visualize_bigraph
+    >>> image = visualize_bigraph(biadjacency)
     >>> image[1:4]
     'svg'
     """
@@ -820,3 +820,220 @@ def svg_bigraph(biadjacency: sparse.csr_matrix,
             f.write(svg)
 
     return svg
+
+
+def svg_graph(adjacency: Optional[sparse.csr_matrix] = None, position: Optional[np.ndarray] = None,
+              names: Optional[np.ndarray] = None, labels: Optional[Iterable] = None, name_position: str = 'right',
+              scores: Optional[Iterable] = None, probs: Optional[Union[np.ndarray, sparse.csr_matrix]] = None,
+              seeds: Union[list, dict] = None, width: Optional[float] = 400, height: Optional[float] = 300,
+              margin: float = 20, margin_text: float = 3, scale: float = 1, node_order: Optional[np.ndarray] = None,
+              node_size: float = 7, node_size_min: float = 1, node_size_max: float = 20,
+              display_node_weight: Optional[bool] = None, node_weights: Optional[np.ndarray] = None,
+              node_width: float = 1, node_width_max: float = 3, node_color: str = 'gray',
+              display_edges: bool = True, edge_labels: Optional[list] = None,
+              edge_width: float = 1, edge_width_min: float = 0.5,
+              edge_width_max: float = 20, display_edge_weight: bool = True,
+              edge_color: Optional[str] = None, label_colors: Optional[Iterable] = None,
+              font_size: int = 12, directed: Optional[bool] = None, filename: Optional[str] = None) -> str:
+    """Return the image of a graph in SVG format.
+
+    Alias for visualize_graph.
+
+    Parameters
+    ----------
+    adjacency :
+        Adjacency matrix of the graph.
+    position :
+        Positions of the nodes.
+    names :
+        Names of the nodes.
+    labels :
+        Labels of the nodes (negative values mean no label).
+    name_position :
+        Position of the names (left, right, above, below)
+    scores :
+        Scores of the nodes (measure of importance).
+    probs :
+        Probability distribution over labels.
+    seeds :
+        Nodes to be highlighted (if dict, only keys are considered).
+    width :
+        Width of the image.
+    height :
+        Height of the image.
+    margin :
+        Margin of the image.
+    margin_text :
+        Margin between node and text.
+    scale :
+        Multiplicative factor on the dimensions of the image.
+    node_order :
+        Order in which nodes are displayed.
+    node_size :
+        Size of nodes.
+    node_size_min :
+        Minimum size of a node.
+    node_size_max:
+        Maximum size of a node.
+    node_width :
+        Width of node circle.
+    node_width_max :
+        Maximum width of node circle.
+    node_color :
+        Default color of nodes (svg color).
+    display_node_weight :
+        If ``True``, display node weights through node size.
+    node_weights :
+        Node weights.
+    display_edges :
+        If ``True``, display edges.
+    edge_labels :
+        Labels of the edges, as a list of tuples (source, destination, label)
+    edge_width :
+        Width of edges.
+    edge_width_min :
+        Minimum width of edges.
+    edge_width_max :
+        Maximum width of edges.
+    display_edge_weight :
+        If ``True``, display edge weights through edge widths.
+    edge_color :
+        Default color of edges (svg color).
+    label_colors:
+        Colors of the labels (svg colors).
+    font_size :
+        Font size.
+    directed :
+        If ``True``, considers the graph as directed.
+    filename :
+        Filename for saving image (optional).
+
+    Returns
+    -------
+    image : str
+        SVG image.
+    """
+    return visualize_graph(adjacency, position, names, labels, name_position, scores, probs, seeds, width, height,
+                           margin, margin_text, scale, node_order, node_size, node_size_min, node_size_max,
+                           display_node_weight, node_weights, node_width, node_width_max, node_color, display_edges,
+                           edge_labels, edge_width, edge_width_min, edge_width_max, display_edge_weight, edge_color,
+                           label_colors, font_size, directed, filename)
+
+
+def svg_bigraph(biadjacency: sparse.csr_matrix,
+                names_row: Optional[np.ndarray] = None, names_col: Optional[np.ndarray] = None,
+                labels_row: Optional[Union[dict, np.ndarray]] = None,
+                labels_col: Optional[Union[dict, np.ndarray]] = None,
+                scores_row: Optional[Union[dict, np.ndarray]] = None,
+                scores_col: Optional[Union[dict, np.ndarray]] = None,
+                probs_row: Optional[Union[np.ndarray, sparse.csr_matrix]] = None,
+                probs_col: Optional[Union[np.ndarray, sparse.csr_matrix]] = None,
+                seeds_row: Union[list, dict] = None, seeds_col: Union[list, dict] = None,
+                position_row: Optional[np.ndarray] = None, position_col: Optional[np.ndarray] = None,
+                reorder: bool = True, width: Optional[float] = 400,
+                height: Optional[float] = 300, margin: float = 20, margin_text: float = 3, scale: float = 1,
+                node_size: float = 7, node_size_min: float = 1, node_size_max: float = 20,
+                display_node_weight: bool = False,
+                node_weights_row: Optional[np.ndarray] = None, node_weights_col: Optional[np.ndarray] = None,
+                node_width: float = 1, node_width_max: float = 3,
+                color_row: str = 'gray', color_col: str = 'gray', label_colors: Optional[Iterable] = None,
+                display_edges: bool = True, edge_labels: Optional[list] = None, edge_width: float = 1,
+                edge_width_min: float = 0.5, edge_width_max: float = 10, edge_color: str = 'black',
+                display_edge_weight: bool = True,
+                font_size: int = 12, filename: Optional[str] = None) -> str:
+    """Return the image of a bipartite graph in SVG format.
+
+    Alias for visualize_bigraph.
+
+    Parameters
+    ----------
+    biadjacency :
+        Biadjacency matrix of the graph.
+    names_row :
+        Names of the rows.
+    names_col :
+        Names of the columns.
+    labels_row :
+        Labels of the rows (negative values mean no label).
+    labels_col :
+        Labels of the columns (negative values mean no label).
+    scores_row :
+        Scores of the rows (measure of importance).
+    scores_col :
+        Scores of the columns (measure of importance).
+    probs_row :
+        Probability distribution over labels for rows.
+    probs_col :
+        Probability distribution over labels for columns.
+    seeds_row :
+        Rows to be highlighted (if dict, only keys are considered).
+    seeds_col :
+        Columns to be highlighted (if dict, only keys are considered).
+    position_row :
+        Positions of the rows.
+    position_col :
+        Positions of the columns.
+    reorder :
+        Use clustering to order nodes.
+    width :
+        Width of the image.
+    height :
+        Height of the image.
+    margin :
+        Margin of the image.
+    margin_text :
+        Margin between node and text.
+    scale :
+        Multiplicative factor on the dimensions of the image.
+    node_size :
+        Size of nodes.
+    node_size_min :
+        Minimum size of nodes.
+    node_size_max :
+        Maximum size of nodes.
+    display_node_weight :
+        If ``True``, display node weights through node size.
+    node_weights_row :
+        Weights of rows (used only if **display_node_weight** is ``True``).
+    node_weights_col :
+        Weights of columns (used only if **display_node_weight** is ``True``).
+    node_width :
+        Width of node circle.
+    node_width_max :
+        Maximum width of node circle.
+    color_row :
+        Default color of rows (svg color).
+    color_col :
+        Default color of cols (svg color).
+    label_colors :
+        Colors of the labels (svg color).
+    display_edges :
+        If ``True``, display edges.
+    edge_labels :
+        Labels of the edges, as a list of tuples (source, destination, label)
+    edge_width :
+        Width of edges.
+    edge_width_min :
+        Minimum width of edges.
+    edge_width_max :
+        Maximum width of edges.
+    display_edge_weight :
+        If ``True``, display edge weights through edge widths.
+    edge_color :
+        Default color of edges (svg color).
+    font_size :
+        Font size.
+    filename :
+        Filename for saving image (optional).
+
+    Returns
+    -------
+    image : str
+        SVG image.
+    """
+    return visualize_bigraph(biadjacency, names_row, names_col, labels_row, labels_col, scores_row, scores_col,
+                             probs_row, probs_col, seeds_row, seeds_col, position_row, position_col, reorder,
+                             width, height, margin, margin_text, scale, node_size, node_size_min, node_size_max,
+                             display_node_weight, node_weights_row, node_weights_col, node_width, node_width_max,
+                             color_row, color_col, label_colors, display_edges, edge_labels, edge_width, edge_width_min,
+                             edge_width_max, edge_color, display_edge_weight, font_size, filename)

sknetwork/visualization/tests/test_dendrograms.py

@@ -1,6 +1,6 @@
 #!/usr/bin/env python3
 # -*- coding: utf-8 -*-
-"""tests for visualization/dendrograms.py"""
+"""tests for visualization of dendrograms"""
 
 import tempfile
 import unittest
@@ -9,7 +9,7 @@ import numpy as np
 
 from sknetwork.data.toy_graphs import karate_club, painters
 from sknetwork.hierarchy import Paris
-from sknetwork.visualization.dendrograms import svg_dendrogram, svg_dendrogram_top
+from sknetwork.visualization.dendrograms import visualize_dendrogram, svg_dendrogram, svg_dendrogram_top
 
 
 # noinspection DuplicatedCode
@@ -22,9 +22,9 @@ class TestVisualization(unittest.TestCase):
         image = svg_dendrogram(dendrogram)
         self.assertEqual(image[1:4], 'svg')
         n = adjacency.shape[0]
-        image = svg_dendrogram(dendrogram, names=np.arange(n), width=200, height=200, margin=10, margin_text=5, scale=3,
-                               n_clusters=3, color='green', colors=['red', 'blue'], font_size=14, reorder=True,
-                               rotate=True)
+        image = visualize_dendrogram(dendrogram, names=np.arange(n), width=200, height=200, margin=10, margin_text=5,
+                                     scale=3, n_clusters=3, color='green', colors=['red', 'blue'], font_size=14,
+                                     reorder=True, rotate=True)
         self.assertEqual(image[1:4], 'svg')
         image = svg_dendrogram(dendrogram, names=np.arange(n), width=200, height=200, margin=10, margin_text=5, scale=3,
                                n_clusters=3, color='green', colors={0: 'red', 1: 'blue'}, font_size=14, reorder=False,
@@ -40,14 +40,14 @@ class TestVisualization(unittest.TestCase):
         names = graph.names
         paris = Paris()
         dendrogram = paris.fit_transform(adjacency)
-        image = svg_dendrogram(dendrogram)
+        image = visualize_dendrogram(dendrogram)
         self.assertEqual(image[1:4], 'svg')
-        image = svg_dendrogram(dendrogram, names=names, width=200, height=200, margin=10, margin_text=5, scale=3,
-                               n_clusters=3, color='green', font_size=14, reorder=True, rotate=True)
+        image = visualize_dendrogram(dendrogram, names=names, width=200, height=200, margin=10, margin_text=5, scale=3,
+                                     n_clusters=3, color='green', font_size=14, reorder=True, rotate=True)
         self.assertEqual(image[1:4], 'svg')
 
         filename = tempfile.gettempdir() + '/image'
-        _ = svg_dendrogram(dendrogram, filename=filename)
+        _ = visualize_dendrogram(dendrogram, filename=filename)
         with open(filename + '.svg', 'r') as f:
             row = f.readline()
             self.assertEqual(row[1:4], 'svg')