Package ai.djl.basicdataset.nlp

Class CookingStackExchange

java.lang.Object

ai.djl.basicdataset.nlp.CookingStackExchange

All Implemented Interfaces:

ai.djl.training.dataset.Dataset, ai.djl.training.dataset.RawDataset<Path>

public class CookingStackExchange
extends Object
implements ai.djl.training.dataset.RawDataset<Path>

A text classification dataset contains questions from cooking.stackexchange.com and their associated tags on the site.

Nested Class Summary

Nested Classes

Modifier and Type	Class	Description
static final class	CookingStackExchange.Builder	A builder to construct a CookingStackExchange.

Nested classes/interfaces inherited from interface ai.djl.training.dataset.Dataset

ai.djl.training.dataset.Dataset.Usage

Method Summary

Modifier and Type	Method	Description
static CookingStackExchange.Builder	builder()	Creates a builder to build a CookingStackExchange.
Path	getData()
Iterable<ai.djl.training.dataset.Batch>	getData(ai.djl.ndarray.NDManager manager)
void	prepare(ai.djl.util.Progress progress)

Methods inherited from class java.lang.Object

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

Methods inherited from interface ai.djl.training.dataset.Dataset

getData, matchingTranslatorOptions, prepare

Method Details

getData

public Path getData()
             throws IOException

Specified by:

getData in interface ai.djl.training.dataset.RawDataset<Path>

Throws:

IOException

getData

public Iterable<ai.djl.training.dataset.Batch> getData(ai.djl.ndarray.NDManager manager)

Specified by:

getData in interface ai.djl.training.dataset.Dataset

prepare

public void prepare(ai.djl.util.Progress progress)
             throws IOException

Specified by:

prepare in interface ai.djl.training.dataset.Dataset

Throws:

IOException

builder

public static CookingStackExchange.Builder builder()

Creates a builder to build a CookingStackExchange.

Returns:

a new builder