Skip to content
Snippets Groups Projects
Commit b2233cc8 authored by Ubuntu's avatar Ubuntu
Browse files

:pencil2: adjust default input and output path

parent d6c4697b
No related branches found
No related tags found
No related merge requests found
......@@ -21,8 +21,8 @@ logging.basicConfig(level=logging.INFO)
def parse_args():
parser = argparse.ArgumentParser(description="Compute pseudo-perplexity for each word in a text.")
parser.add_argument("-m", "--model-name", type=str, default="bert-base-multilingual-uncased", help="Model name or path")
parser.add_argument("-i", "--input-path", type=str, default="data/sentences", help="Path to the input directory")
parser.add_argument("-o", "--output-path", type=str, default="data/pppl_per_sent", help="Path to the output directory")
parser.add_argument("-i", "--input-path", type=str, default="data/worderror_json", help="Path to the input directory")
parser.add_argument("-o", "--output-path", type=str, default="data/pppl", help="Path to the output directory")
parser.add_argument("-w", "--window-size", type=int, default=11, help="Size of the sliding window")
parser.add_argument("-b", "--batch-size", type=int, default=32, help="Batch size for processing")
return parser.parse_args()
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment