TrainingListener.BatchData (Deep Java Library 0.24.0 API specification)

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	Description
`BatchData(Batch batch, java.util.Map<Device,NDList> labels, java.util.Map<Device,NDList> predictions)`	Constructs a new `TrainingListener.BatchData`.

Method Summary

All Methods Instance Methods Concrete Methods
Modifier and Type	Method	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