prepare.py 文件源码

def main():
    np.random.seed(1234)

    workdir = sys.argv[1]
    if len(sys.argv) > 2:
        snapshot = int(sys.argv[2])
    else:
        snapshot = -1
    if len(sys.argv) > 3:
        embed_size = int(sys.argv[3])
    else:
        embed_size = 300

    dataset = os.getenv('DATASET', workdir)
    glove = os.getenv('GLOVE', os.path.join(workdir, 'glove.42B.300d.txt'))

    download_glove(glove)

    input_words = set()
    # add the canonical words for the builtin functions
    add_words(input_words, 'now nothing notify return the event')

    create_dictionary(input_words, dataset)
    get_thingpedia(input_words, workdir, snapshot)
    save_dictionary(input_words, workdir)
    trim_embeddings(input_words, workdir, embed_size, glove)