Skip to content

Commit f5c3c1b

Browse files
committed
args
1 parent c6764b3 commit f5c3c1b

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

python/trainXGBoostforDirection.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -391,18 +391,18 @@ def main():
391391
parser = argparse.ArgumentParser(
392392
description=("Train XGBoost Multi-Target BDTs for Direction Reconstruction")
393393
)
394-
parser.add_argument("input_file_list", help="List of input mscw ROOT files.")
395-
parser.add_argument("ntel", type=int, help="Telescope multiplicity (2, 3, or 4).")
394+
parser.add_argument("--input_file_list", help="List of input mscw ROOT files.")
395+
parser.add_argument("--ntel", type=int, help="Telescope multiplicity (2, 3, or 4).")
396396
parser.add_argument(
397397
"output_dir", help="Output directory for XGBoost models and weights."
398398
)
399399
parser.add_argument(
400-
"train_test_fraction",
400+
"--train_test_fraction",
401401
type=float,
402402
help="Fraction of data for training (e.g., 0.5).",
403403
)
404404
parser.add_argument(
405-
"max_events",
405+
"--max_events",
406406
type=int,
407407
help="Maximum number of events to process across all files.",
408408
)

0 commit comments

Comments
 (0)