ddrous

import dgl
import networkx as nx
from pyvis.network import Network

# Define number of nodes to show in visualization
nb_nodes_plot = 100

# Step 1. Load the cora dataset and slice it using DGL
dataset = dgl.data.CoraGraphDataset()
g_dgl = dataset[0].subgraph(list(range(nb_nodes_plot)))

ddrous

#%%
import jax
import jax.numpy as jnp
import numpy as np
jax.numpy.set_printoptions(precision=6)



# %% [markdown]