def get_eval_KB_completion_parser()

in lama/options.py [0:0]

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

def get_eval_KB_completion_parser():
    parser = get_general_parser()
    parser.add_argument(
        "--dataset-filename",
        "--df",
        dest="dataset_filename",
        help="filename containing dataset",
    )
    parser.add_argument(
        "--logdir",
        dest="logdir",
        default="../experiments_logs/",
        help="logging directory",
    )
    parser.add_argument(
        "--full-logdir",
        help="Full path to the logging folder. If set, wiill override log_dir.",
    )
    parser.add_argument(
        "--template", dest="template", default="", help="template for surface relation"
    )
    parser.add_argument(
        "--batch-size", dest="batch_size", type=int, default=32, help="batch size"
    )
    parser.add_argument(
        "--lowercase",
        "--lower",
        dest="lowercase",
        action="store_true",
        help="perform the evaluation using lowercase text",
    )
    parser.add_argument(
        "--threads",
        dest="threads",
        type=int,
        default=-1,
        help="number of threads for evaluation metrics computation (defaults: all available)",
    )
    return parser