make graph visualization become colorful
This commit is contained in:
parent
c800fa4843
commit
f400b02b0f
1 changed files with 5 additions and 0 deletions
|
|
@ -1,5 +1,6 @@
|
|||
import networkx as nx
|
||||
from pyvis.network import Network
|
||||
import random
|
||||
|
||||
# Load the GraphML file
|
||||
G = nx.read_graphml('./dickens/graph_chunk_entity_relation.graphml')
|
||||
|
|
@ -10,5 +11,9 @@ net = Network(notebook=True)
|
|||
# Convert NetworkX graph to Pyvis network
|
||||
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
|
||||
net.show('knowledge_graph.html')
|
||||
Loading…
Add table
Reference in a new issue