Source code for pygsp.graphs.fullconnected

# -*- coding: utf-8 -*-

from . import Graph

import numpy as np


class FullConnected(Graph):
    r"""
    Create a fully connected graph.

    Parameters
    ----------
    N : int
        Number of vertices (default = 10)

    Examples
    --------
    >>> from pygsp import graphs
    >>> G = graphs.FullConnected(N=5)

    """

    def __init__(self, N=10):

        tmp = np.arange(N).reshape(N, 1)

        W = np.ones((N, N)) - np.identity(N)
        plotting = {'limits': np.array([-1, 1, -1, 1])}

        super(FullConnected, self).__init__(W=W, gtype='full', plotting=plotting)