From b5add710e574160b162ee1f17f29f029d8d0be80 Mon Sep 17 00:00:00 2001 From: james szalkie Date: Thu, 13 Aug 2026 12:42:13 -0400 Subject: [PATCH] Training builder --- Armory/TrainingBuilder.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/Armory/TrainingBuilder.py b/Armory/TrainingBuilder.py index 8ccb613..e360b5d 100644 --- a/Armory/TrainingBuilder.py +++ b/Armory/TrainingBuilder.py @@ -32,7 +32,7 @@ from tensorflow.keras import Sequential from tensorflow.keras.layers import Dense from tensorflow.keras.callbacks import EarlyStopping -ROOT_FOLDER = Path("/Users/jamesszalkie/ANASEN_analysis/Armory/Training_Data_Ne/") +ROOT_FOLDER = Path("/Users/jamesszalkie/ANASEN_analysis/Armory/Training_Data/") TREE_NAME = "tree1" @@ -44,6 +44,10 @@ INPUT_BRANCHES = [ "MTarget", "MLight", "MHeavy", + "ZBeam", + "ZHeavy", + "ZLight", + "ZTarget" ] OUTPUT_BRANCHES = [