hypgraph.py 文件源码

def _render(self, costs=False, word_probs=False, highlight_best=False):
        from pygraphviz import AGraph
        graph = AGraph(directed=True)
        for node_id, node_label in self.nodes.iteritems():
            attributes = self._node_attr(node_id, costs=costs, word_probs=word_probs)
            graph.add_node(node_id, **attributes)
        for (parent_node_id, child_node_id) in self.edges:
            graph.add_edge(parent_node_id, child_node_id)
        self.graph = graph
        if highlight_best:
            self._highlight_best()