9df8cd324207d3924439c3d912229811e34687f0,python/dgl/model_zoo/chem/pretrain.py,,load_pretrained,#Any#Any#,58

Before Change

        model = MPNNModel(output_dim=12)

    elif model_name == "JTNN_ZINC":
        vocab_file = "{}/jtnn/{}.txt".format(get_download_dir(), "vocab")
        model = DGLJTNNVAE(vocab_file=vocab_file,
                           depth=3,
                           hidden_size=450,
                           latent_size=56)

After Change

        vocab_file = "{}/jtnn/{}.txt".format(default_dir, "vocab")
        if not os.path.exists(vocab_file):
            zip_file_path = "{}/jtnn.zip".format(default_dir)
            download("https://s3-ap-southeast-1.amazonaws.com/dgl-data-cn/dataset/jtnn.zip",
                     path=zip_file_path)
            extract_archive(zip_file_path, "{}/jtnn".format(default_dir))
        model = DGLJTNNVAE(vocab_file=vocab_file,
                           depth=3,
                           hidden_size=450,

Instances