rtang/ANASEN_analysis
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Training builder

Browse Source
This commit is contained in:
James Szalkie 2026-08-13 12:42:13 -04:00
parent 7181b80635
commit b5add710e5
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
Armory/TrainingBuilder.py
View File

@ -32,7 +32,7 @@ from tensorflow.keras import Sequential
from tensorflow.keras.layers import Dense from tensorflow.keras.layers import Dense
from tensorflow.keras.callbacks import EarlyStopping 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" TREE_NAME = "tree1"
@ -44,6 +44,10 @@ INPUT_BRANCHES = [
"MTarget", "MTarget",
"MLight", "MLight",
"MHeavy", "MHeavy",
"ZBeam",
"ZHeavy",
"ZLight",
"ZTarget"
] ]
OUTPUT_BRANCHES = [ OUTPUT_BRANCHES = [