INNER CODE UNIT · Python
__init__
IBM/transition-amr-parser · src/ibm_neural_aligner/main.py:441
def __init__(self, corpus, text_tokenizer=None, amr_tokenizer=None):
self.corpus = corpus
self.text_tokenizer = text_tokenizer
self.amr_tokenizer = amr_tokenizer
self.cached = {}
def get_dgl_graph(self, amr):
vocab = self.amr_tokenizer.vocab
# init graph
g = dgl.DGLGraph()
# get tree
tree = convert_amr_to_tree(amr)
# add node structure
node_ids = get_node_ids(amr)
node_TO_idx = {k: i for i, k in enumerate(node_ids)}