Package ai.djl.basicdataset.tabular

Class AmesRandomAccess.Builder

java.lang.Object

ai.djl.training.dataset.RandomAccessDataset.BaseBuilder<T>

ai.djl.basicdataset.tabular.TabularDataset.BaseBuilder<T>

ai.djl.basicdataset.tabular.CsvDataset.CsvBuilder<AmesRandomAccess.Builder>

ai.djl.basicdataset.tabular.AmesRandomAccess.Builder

Enclosing class:

AmesRandomAccess

public static final class AmesRandomAccess.Builder
extends CsvDataset.CsvBuilder<AmesRandomAccess.Builder>

A builder to construct a AmesRandomAccess.

Field Summary

Fields inherited from class ai.djl.basicdataset.tabular.CsvDataset.CsvBuilder

csvFormat, csvUrl

Fields inherited from class ai.djl.basicdataset.tabular.TabularDataset.BaseBuilder

features, labels

Fields inherited from class ai.djl.training.dataset.RandomAccessDataset.BaseBuilder

dataBatchifier, device, labelBatchifier, limit, pipeline, prefetchNumber, sampler, targetPipeline

Method Summary

Modifier and Type	Method	Description
AmesRandomAccess.Builder	addFeature(java.lang.String name)	Adds a feature to the features set.
AmesRandomAccess.Builder	addFeature(java.lang.String name, boolean onehotEncode)	Adds a feature to the features set with onehot encoding.
AmesRandomAccess	build()	Builds the new AmesRandomAccess.
java.util.List<java.lang.String>	getAvailableFeatures()	Returns the available features of this dataset.
AmesRandomAccess.Builder	optArtifactId(java.lang.String artifactId)	Sets the optional artifactId.
AmesRandomAccess.Builder	optGroupId(java.lang.String groupId)	Sets optional groupId.
AmesRandomAccess.Builder	optRepository(ai.djl.repository.Repository repository)	Sets the optional repository.
AmesRandomAccess.Builder	optUsage(ai.djl.training.dataset.Dataset.Usage usage)	Sets the optional usage.
AmesRandomAccess.Builder	self()

Methods inherited from class ai.djl.basicdataset.tabular.CsvDataset.CsvBuilder

optCsvFile, optCsvUrl, setCsvFormat

Methods inherited from class ai.djl.basicdataset.tabular.TabularDataset.BaseBuilder

addCategoricalFeature, addCategoricalFeature, addCategoricalFeature, addCategoricalLabel, addCategoricalLabel, addCategoricalLabel, addFeature, addLabel, addNumericFeature, addNumericFeature, addNumericLabel, addNumericLabel, validate

Methods inherited from class ai.djl.training.dataset.RandomAccessDataset.BaseBuilder

addTargetTransform, addTransform, getSampler, optDataBatchifier, optDevice, optLabelBatchifier, optLimit, optPipeline, optPrefetchNumber, optTargetPipeline, setSampling, setSampling, setSampling

Methods inherited from class java.lang.Object

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

Method Detail

self

public AmesRandomAccess.Builder self()

Overrides:

self in class CsvDataset.CsvBuilder<AmesRandomAccess.Builder>

optUsage

public AmesRandomAccess.Builder optUsage(ai.djl.training.dataset.Dataset.Usage usage)

Sets the optional usage.

Parameters:

usage - the new usage

Returns:

this builder

optRepository

public AmesRandomAccess.Builder optRepository(ai.djl.repository.Repository repository)

Sets the optional repository.

Parameters:

repository - the repository

Returns:

this builder

optGroupId

public AmesRandomAccess.Builder optGroupId(java.lang.String groupId)

Sets optional groupId.

Parameters:

groupId - the groupId}

Returns:

this builder

optArtifactId

public AmesRandomAccess.Builder optArtifactId(java.lang.String artifactId)

Sets the optional artifactId.

Parameters:

artifactId - the artifactId

Returns:

this builder

addFeature

public AmesRandomAccess.Builder addFeature(java.lang.String name)

Adds a feature to the features set.

Parameters:

name - the name of the feature

Returns:

this builder

addFeature

public AmesRandomAccess.Builder addFeature(java.lang.String name,
                                           boolean onehotEncode)

Adds a feature to the features set with onehot encoding.

Parameters:

name - the name of the feature

onehotEncode - true if use onehot encoding

Returns:

this builder

getAvailableFeatures

public java.util.List<java.lang.String> getAvailableFeatures()

Returns the available features of this dataset.

Returns:

a list of feature names

build

public AmesRandomAccess build()

Builds the new AmesRandomAccess.

Overrides:

build in class CsvDataset.CsvBuilder<AmesRandomAccess.Builder>

Returns:

the new AmesRandomAccess