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 java.lang.Object

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

Constructor Summary

Constructors

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

Method Summary

Modifier and Type	Method and Description
Batch	getBatch() Returns the original batch.
java.util.Map<Device,NDList>	getLabels() Returns the labels for each device.
java.util.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 Detail

BatchData

public BatchData(Batch batch,
                 java.util.Map<Device,NDList> labels,
                 java.util.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 Detail

getBatch

public Batch getBatch()

Returns the original batch.

Returns:

the original batch

getLabels

public java.util.Map<Device,NDList> getLabels()

Returns the labels for each device.

Returns:

the labels for each device

getPredictions

public java.util.Map<Device,NDList> getPredictions()

Returns the predictions for each device.

Returns:

the predictions for each device