def parse_args()

in src/run_paraphrase.py [0:0]

• 13 lines of code
• 1 McCabe index (conditional complexity)

def parse_args(args=None):
    parser = argparse.ArgumentParser('Run paraphrase detection with BERTScore experiments.')
    parser.add_argument('dataset', choices=['qqp', 'mrpc', 'paws-wiki', 'paws-qqp'], help='Which dataset to use')
    parser.add_argument('load_dir', help='Directory containing RoBERTa checkpoints')
    parser.add_argument('--batch-size', '-b', default=16, type=int, help='Maximum batch size')
    # Fine-tuning 
    parser.add_argument('--train-ft', action='store_true', help='Train with fine-tuning.')
    parser.add_argument('--learning-rate', '-r', type=float, default=1e-5)
    parser.add_argument('--logit-scale', '-l', type=float, default=1e3)
    parser.add_argument('--num-epochs', '-T', type=int, default=20)
    parser.add_argument('--rng-seed', type=int, default=0)
    # scikit-learn training
    parser.add_argument('--train-skl', action='store_true', help='Train a scikit-learn model')
    parser.add_argument('--skl-C', type=float, default=1.0)
    return parser.parse_args(args=args)