Class ModelTrackFinding

java.lang.Object

org.jlab.rec.ahdc.AI.ModelTrackFinding

public class ModelTrackFinding
extends Object

Model of What. \todo fix class name

Constructor Summary

Constructors

Constructor	Description
ModelTrackFinding()

Method Summary

Modifier and Type	Method	Description
float[]	batchPredict(float[][] inputs)	Batch prediction for improved performance.
ai.djl.repository.zoo.ZooModel<float[],Float>	getModel()

Methods inherited from class Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Details

ModelTrackFinding

public ModelTrackFinding()

Method Details

getModel

public ai.djl.repository.zoo.ZooModel<float[],Float> getModel()

batchPredict

public float[] batchPredict(float[][] inputs)
                     throws Exception

Batch prediction for improved performance. Predicts all tracks at once instead of one at a time. This is significantly faster due to reduced overhead and better GPU utilization.

Parameters:

inputs - Array of input features for each track

Returns:

Array of predictions for each track

Throws:

Exception