bf

rkansal47 · Nov 3, 2023 · 0db8cd0 · 0db8cd0
1 parent f3367b5
commit 0db8cd0
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/HHbbVV/postprocessing/TrainBDT.py b/src/HHbbVV/postprocessing/TrainBDT.py
@@ -355,7 +355,9 @@ def main(args):
         )
 
     if not args.inference_only:
-        evaluate_model(model, args.model_dir, train, test, multiclass=args.multiclass)
+        evaluate_model(
+            model, args.model_dir, train, test, args.test_size, multiclass=args.multiclass
+        )
 
     if not args.evaluate_only:
         do_inference(model, args.model_dir, data_dict, multiclass=args.multiclass)