INNER CODE UNIT · Python

parse_args

yuhaozhang/tacred-relation · ensemble.py:13

def parse_args():
    parser = argparse.ArgumentParser()
    parser.add_argument('pred_files', nargs='+', help='A list of prediction files written by eval.py.')
    parser.add_argument('--data_dir', default='dataset/tacred')
    parser.add_argument('--dataset', default='test', help='Evaluate on dev or test set.')
    args = parser.parse_args()
    return args

def main():
    args = parse_args()
    print("Loading data file...")
    filename = args.data_dir + '/{}.json'.format(args.dataset)
    with open(filename, 'r') as infile:
        data = json.load(infile, encoding='utf8')
    labels = [d['relation'] for d in data]

    # read predictions
    print("Loading {} prediction files...".format(len(args.pred_files)))

View source record →