INNER CODE UNIT · Python

Dataset

IBM/transition-amr-parser · src/ibm_neural_aligner/main.py:440

class Dataset(object):
    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)

View source record →