def plot_adjacency_matrix(self, im=None, ax=None, cmap="Reds", vmax=None):
        import matplotlib.pyplot as plt
        if vmax is None:
            vmax = np.max(self.W_effective)

        if im is None:
            if ax is None:
                ax = plt.gca()

            im = ax.imshow(self.W_effective, interpolation="none", cmap=cmap, vmin=0, vmax=vmax)
            ax.set_ylabel('k')
            ax.set_xlabel('k\'')
            ax.set_title('W_{k \\to k\'}')

        else:
            # Update given image
            im.set_data(self.W_effective)

        return im
dataset/ETHPy150Open slinderman/pyhawkes/pyhawkes/models.py/_DiscreteTimeNetworkHawkesModelBase.plot_adjacency_matrix/original