Skip to content
Snippets Groups Projects
FactorTree.py 319 B
Newer Older
Denis John PC's avatar
Denis John PC committed

import networkx as nx



class FactorTree(object):
    
    
    def __init__(self,graph,rootNode):
        self.graph = graph
        self.rootNode = rootNode
        
Denis John PC's avatar
Denis John PC committed
    def draw(self):
        import matplotlib.pyplot as plt
        nx.draw_circular(self.graph)
        plt.show()