Package ai.djl.training.listener

Class TrainingListener.BatchData

java.lang.Object

ai.djl.training.listener.TrainingListener.BatchData

Enclosing interface:

TrainingListener

public static class TrainingListener.BatchData
extends Object

A class to pass data from the batch into the training listeners.

Constructor Summary

Constructors

Constructor	Description
BatchData(Batch batch, Map<Device,NDList> labels, Map<Device,NDList> predictions)	Constructs a new TrainingListener.BatchData.

Method Summary

Modifier and Type	Method	Description
Batch	getBatch()	Returns the original batch.
Map<Device,NDList>	getLabels()	Returns the labels for each device.
Map<Device,NDList>	getPredictions()	Returns the predictions for each device.

Methods inherited from class java.lang.Object

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

Constructor Details

BatchData

public BatchData(Batch batch,
 Map<Device,NDList> labels,
 Map<Device,NDList> predictions)

Constructs a new TrainingListener.BatchData.

Parameters:

batch - the original batch

labels - the labels for each device

predictions - the predictions for each device

Method Details

getBatch

public Batch getBatch()

Returns the original batch.

Returns:

the original batch

getLabels

public Map<Device,NDList> getLabels()

Returns the labels for each device.

Returns:

the labels for each device

getPredictions

public Map<Device,NDList> getPredictions()

Returns the predictions for each device.

Returns:

the predictions for each device