gardel/lightrag
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

make graph visualization become colorful

Browse Source
This commit is contained in:
nongbin
2024-10-20 21:17:09 +08:00
parent 6be119c2dd
commit 1b45fe26c7
1 changed files with 5 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
examples/graph_visual.py
View File

@@ -1,5 +1,6 @@
import networkx as nx import networkx as nx
from pyvis.network import Network from pyvis.network import Network
import random
# Load the GraphML file # Load the GraphML file
G = nx.read_graphml('./dickens/graph_chunk_entity_relation.graphml') G = nx.read_graphml('./dickens/graph_chunk_entity_relation.graphml')
@@ -10,5 +11,9 @@ net = Network(notebook=True)
# Convert NetworkX graph to Pyvis network # Convert NetworkX graph to Pyvis network
net.from_nx(G) net.from_nx(G)
# Add colors to nodes
for node in net.nodes:
node['color'] = "#{:06x}".format(random.randint(0, 0xFFFFFF))
# Save and display the network # Save and display the network
net.show('knowledge_graph.html') net.show('knowledge_graph.html')