onmt.Dataset

Data management and batch creation. Handles data created by preprocess.lua.

[src]

onmt.Dataset(srcData, tgtData)

Initialize a data object given aligned tables of IntTensors srcData and tgtData.

[src]

onmt.Dataset:setBatchSize(maxBatchSize)

Setup up the training data to respect maxBatchSize.

[src]

onmt.Dataset:batchCount()

Return number of batches.

[src]

onmt.Dataset:getBatch(idx)

Get Batch number idx. If nil make a batch of all the data.