ai.djl.training

Class DataManager

java.lang.Object

ai.djl.training.DataManager

public class DataManager
extends java.lang.Object

DataManager is an class that is used primarily in the Trainer to manipulate data from the dataset to be appropriate for the current model.

Field Summary

Fields

Modifier and Type	Field and Description
static DataManager	DEFAULT_DATA_MANAGER

Constructor Summary

Constructors

Constructor and Description
DataManager()

Method Summary

Modifier and Type	Method and Description
NDList	getData(Batch batch) Fetches data from the given Batch in required form.
NDList	getLabels(Batch batch) Fetches labels from the given Batch in required form.

Methods inherited from class java.lang.Object

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

Field Detail

DEFAULT_DATA_MANAGER

public static final DataManager DEFAULT_DATA_MANAGER

Constructor Detail

DataManager

public DataManager()

Method Detail

getData

public NDList getData(Batch batch)

Fetches data from the given Batch in required form.

Parameters:

batch - the Batch to fetch data from

Returns:

an NDList with the appropriate data

getLabels

public NDList getLabels(Batch batch)

Fetches labels from the given Batch in required form.

Parameters:

batch - the Batch to fetch labels from

Returns:

an NDList with the appropriate labels