Skip to content

Commit 9673941

Browse files
Add an engine to apply new AI models for forward track finding (#1039)
* add engine to apply a new AI model for DC cluster combo prediction * change deafult threshold * Switch to generic predictor pool (#1040) * add generic predictor pool * cleanup --------- Co-authored-by: Nathan Baltzell <[email protected]>
1 parent 4bad064 commit 9673941

File tree

5 files changed

+658
-20
lines changed

5 files changed

+658
-20
lines changed
Lines changed: 39 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,39 @@
1+
package org.jlab.rec.ai.dcCluster;
2+
3+
4+
public class DCCluster{
5+
int id;
6+
int sector;
7+
int superlayer;
8+
float avgWire;
9+
float fitSlope;
10+
11+
public DCCluster(int id, int sector, int superlayer, float avgWire, float fitSlope){
12+
this.id = id;
13+
this.sector = sector;
14+
this.superlayer = superlayer;
15+
this.avgWire = avgWire;
16+
this.fitSlope = fitSlope;
17+
}
18+
19+
public int getId(){
20+
return id;
21+
}
22+
23+
public int getSector(){
24+
return sector;
25+
}
26+
27+
public int getSuperlayer(){
28+
return superlayer;
29+
}
30+
31+
public float getAvgWire(){
32+
return avgWire;
33+
}
34+
35+
public float getFitSlope(){
36+
return fitSlope;
37+
}
38+
39+
}
Lines changed: 40 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,40 @@
1+
package org.jlab.rec.ai.dcCluster;
2+
3+
import java.util.List;
4+
import java.util.ArrayList;
5+
6+
public class DCClusterCombo extends ArrayList<DCCluster> {
7+
8+
private int id = -1;
9+
private int missingSL = -1;
10+
private float probability = -1;
11+
12+
public DCClusterCombo(List<DCCluster> clsList) {
13+
super(clsList);
14+
}
15+
16+
public DCClusterCombo(List<DCCluster> clsList, int missingSL) {
17+
super(clsList);
18+
this.missingSL = missingSL;
19+
}
20+
21+
public int getMissingSL() {
22+
return missingSL;
23+
}
24+
25+
public void setProbability(float probability){
26+
this.probability = probability;
27+
}
28+
29+
public float getProbability(){
30+
return probability;
31+
}
32+
33+
public void setId(int id){
34+
this.id = id;
35+
}
36+
37+
public int getId(){
38+
return id;
39+
}
40+
}

0 commit comments

Comments
 (0)