Skip to content

Instantly share code, notes, and snippets.

@rkreddyp

rkreddyp/kb_visualize.py

Created September 22, 2023 09:54
Show Gist options

  • Select an option

    Learn more about clone URLs
  • Save rkreddyp/a4b5d549e6884ee5a0d88e37562e4890 to your computer and use it in GitHub Desktop.

Select an option

Learn more about clone URLs
Save rkreddyp/a4b5d549e6884ee5a0d88e37562e4890 to your computer and use it in GitHub Desktop.
Download ZIP

Revisions

  1. rkreddyp created this gist Sep 22, 2023.
    15 changes: 15 additions & 0 deletions kb_visualize.py
    View file
    Original file line number Diff line number Diff line change
    @@ -0,0 +1,15 @@
    def visualize_knowledge_graph(kg):
    print (kg.keys())
    dot = Digraph(comment="Knowledge Graph")

    # Add nodes
    for node in kg['nodes']:
    dot.node(str(node['id']), node['label'], color=node['color'])

    # Add edges
    for edge in kg['edges']:
    #dot.edge(str(edge['source']), str(edge['target']), label=edge['label'], color=edge['color'])
    dot.edge(str(edge['source']), str(edge['target']), label=edge['label'])

    # Render the graph
    dot.render("/tmp/knowledge_graph.gv", view=False)