Download.py

from LInK.DataUtils import download_data
from LInK.nn import download_checkpoint, download_emdedding

import argparse

# Set up the argument parser
parser = argparse.ArgumentParser(description='Download the data and the checkpoint')
parser.add_argument('--data_folder', type=str, default='./Data/', help='The folder to store the data')
parser.add_argument('--test_folder', type=str, default='./TestData/', help='The folder to store the test data')
parser.add_argument('--checkpoint_folder', type=str, default='./Checkpoints/', help='The folder to store the checkpoint')
parser.add_argument('--checkpoint_name', type=str, default='checkpoint.LInK', help='The name of the checkpoint file')
parser.add_argument('--gdrive_id', type=str, default='1lNFz8jNrKWeJNqYPuz9LUDF0cb4fl08d', help='The Google Drive ID of the checkpoint file')
parser.add_argument('--download_embedding', type=bool, default=True, help='Whether to download the embedding file')
parser.add_argument('--embedding_folder', type=str, default='./Embedding/', help='The folder to store the embedding file')
args = parser.parse_args()

# download_data(args.data_folder, args.test_folder)
download_checkpoint(args.checkpoint_folder, args.checkpoint_name, args.gdrive_id)

# if args.download_embedding:
    # download_emdedding(embeddings_folder=args.embedding_folder)