Update AHDCEngine to initialize model conditionally and adjust prediction threshold

mathieuouillon · baltzell · commit 3f00dfa2d460 · 2025-03-27T15:20:11.000-04:00
diff --git a/reconstruction/alert/src/main/java/org/jlab/rec/service/AHDCEngine.java b/reconstruction/alert/src/main/java/org/jlab/rec/service/AHDCEngine.java
@@ -57,7 +57,9 @@ public boolean init() {
 
 		}
 
-		model = new Model();
+		if (mode == Mode.AI_Track_Finding) {
+			model = new Model();
+		}
 
 		return true;
 	}
@@ -155,7 +157,7 @@ public int compare(Hit a1, Hit a2) {
 				}
 
 				for (TrackPrediction t : predictions) {
-					if (t.getPrediction() > 0.5)
+					if (t.getPrediction() > 0.2)
 						AHDC_Tracks.add(new Track(t.getClusters()));
 				}
 			}

Original file line number	Diff line number	Diff line change
`@@ -57,7 +57,9 @@ public boolean init() {`
`57`	`57`
`58`	`58`	`}`
`59`	`59`
`60`		`- model = new Model();`
	`60`	`+ if (mode == Mode.AI_Track_Finding) {`
	`61`	`+ model = new Model();`
	`62`	`+ }`
`61`	`63`
`62`	`64`	`return true;`
`63`	`65`	`}`
`@@ -155,7 +157,7 @@ public int compare(Hit a1, Hit a2) {`
`155`	`157`	`}`
`156`	`158`
`157`	`159`	`for (TrackPrediction t : predictions) {`
`158`		`- if (t.getPrediction() > 0.5)`
	`160`	`+ if (t.getPrediction() > 0.2)`
`159`	`161`	`AHDC_Tracks.add(new Track(t.getClusters()));`
`160`	`162`	`}`
`161`	`163`	`}`