Package ai.djl.basicdataset.tabular

Class CsvDataset.CsvBuilder<T extends CsvDataset.CsvBuilder<T>>

java.lang.Object

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

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

ai.djl.basicdataset.tabular.CsvDataset.CsvBuilder<T>

Direct Known Subclasses:

AirfoilRandomAccess.Builder, AmazonReview.Builder, AmesRandomAccess.Builder, DailyDelhiClimate.Builder, MovieLens100k.Builder

Enclosing class:

CsvDataset

public static class CsvDataset.CsvBuilder<T extends CsvDataset.CsvBuilder<T>>
extends TabularDataset.BaseBuilder<T>

Used to build a CsvDataset.

Field Summary

Fields

Modifier and Type	Field	Description
protected org.apache.commons.csv.CSVFormat	csvFormat
protected URL	csvUrl

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

allowNoLabels, features, labels

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

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

Constructor Summary

Constructors

Constructor	Description
CsvBuilder()

Method Summary

Modifier and Type	Method	Description
CsvDataset	build()	Builds the new CsvDataset.
T	optCsvFile(Path csvFile)	Sets the optional CSV file path.
T	optCsvUrl(String csvUrl)	Sets the optional CSV file URL.
protected T	self()
T	setCsvFormat(org.apache.commons.csv.CSVFormat csvFormat)	Sets the CSV file format.

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

addCategoricalFeature, addCategoricalFeature, addCategoricalFeature, addCategoricalLabel, addCategoricalLabel, addCategoricalLabel, addFeature, addLabel, addNumericFeature, addNumericFeature, addNumericLabel, addNumericLabel, noLabels

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

Field Details

csvUrl

protected URL csvUrl

csvFormat

protected org.apache.commons.csv.CSVFormat csvFormat

Constructor Details

CsvBuilder

public CsvBuilder()

Method Details

self

protected T self()

Specified by:

self in class ai.djl.training.dataset.RandomAccessDataset.BaseBuilder<T extends CsvDataset.CsvBuilder<T>>

optCsvFile

public T optCsvFile(Path csvFile)

Sets the optional CSV file path.

Parameters:

csvFile - the CSV file path

Returns:

this builder

optCsvUrl

public T optCsvUrl(String csvUrl)

Sets the optional CSV file URL.

Parameters:

csvUrl - the CSV file URL

Returns:

this builder

setCsvFormat

public T setCsvFormat(org.apache.commons.csv.CSVFormat csvFormat)

Sets the CSV file format.

Parameters:

csvFormat - the CSVFormat

Returns:

this builder

build

public CsvDataset build()

Builds the new CsvDataset.

Returns:

the new CsvDataset