feat: Size arg

2025-12-10 16:29:39 +01:00 · 2025-12-10 16:29:39 +01:00 · 67310ce4f4
commit 67310ce4f4
parent 6c5908e6ae
2 changed files with 6 additions and 1 deletions
--- a/src/args.py
+++ b/src/args.py
@ -35,6 +35,8 @@ def parse_arguments():
    train_parser.add_argument("--method",
                              choices=["fetch", "optuna", "full"], required=True,
                              help="Method to use for training")
+    train_parser.add_argument("--size", "-s", type=int, required=False,
+                              help="Size of the subset of the dataset to use")

    compress_parser = subparsers.add_parser("compress", parents=[modelparser, fileparser])