File size: 1,579 Bytes
4f08d2c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 |
from argparse import ArgumentParser
def get_args():
"""
Parses command-line arguments for SentencePerturbation.
Returns:
argparse.Namespace: Parsed arguments.
"""
parser = ArgumentParser()
parser.add_argument(
"--perturb_dataset",
dest="perturb_dataset",
required=True,
help="Name of the CSV file"
)
parser.add_argument(
"--task",
dest="task",
required=True,
choices=["anto", "jumbling", "syn", "paraphrase"],
help="Task to perform: anto/jumbling/syn/paraphrase",
)
parser.add_argument(
"--M",
dest="model_name",
required=True,
help="LLM Model")
parser.add_argument(
"--target_lang",
dest="target_lang",
required=True,
default="en",
help="Language for translation"
)
parser.add_argument(
"--save",
dest="save",
action="store_true",
help="Save the results in a CSV file",
)
parser.add_argument(
"--gpu",
dest="gpu",
default="auto",
help="GPU to run the model"
)
parser.add_argument(
"--batch_size",
dest="batch_size",
type=int,
default=16,
help="Batch size for translation"
)
parser.add_argument(
"--metric",
dest="metric",
type=str,
default="cosine",
choices=["cosine","ned","both"],
help="Metric to use for comparison",
)
return parser.parse_args()
|