Package ai.djl.basicdataset.tabular

Class CsvDataset

java.lang.Object

ai.djl.training.dataset.RandomAccessDataset

ai.djl.basicdataset.tabular.TabularDataset

ai.djl.basicdataset.tabular.CsvDataset

All Implemented Interfaces:

ai.djl.training.dataset.Dataset

Direct Known Subclasses:

AirfoilRandomAccess, AmazonReview, AmesRandomAccess, DailyDelhiClimate, MovieLens100k

public class CsvDataset
extends TabularDataset

CsvDataset represents the dataset that stored in a .csv file.

Nested Class Summary

Nested Classes

Modifier and Type	Class	Description
static class	CsvDataset.CsvBuilder<T extends CsvDataset.CsvBuilder<T>>	Used to build a CsvDataset.

Nested classes/interfaces inherited from class ai.djl.basicdataset.tabular.TabularDataset

TabularDataset.BaseBuilder<T extends TabularDataset.BaseBuilder<T>>

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

ai.djl.training.dataset.Dataset.Usage

Field Summary

Fields

Modifier and Type	Field	Description
protected org.apache.commons.csv.CSVFormat	csvFormat
protected java.util.List<org.apache.commons.csv.CSVRecord>	csvRecords
protected java.net.URL	csvUrl

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

features, labels

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

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

Constructor Summary

Constructors

Modifier	Constructor	Description
protected	CsvDataset(CsvDataset.CsvBuilder<?> builder)

Method Summary

Modifier and Type	Method	Description
protected long	availableSize()
static CsvDataset.CsvBuilder<?>	builder()	Creates a builder to build a AmesRandomAccess.
java.lang.String	getCell(long rowIndex, java.lang.String featureName)	Returns a cell in the dataset.
java.util.List<java.lang.String>	getColumnNames()	Returns the column names of the CSV file.
void	prepare(ai.djl.util.Progress progress)

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

get, getFeatures, getFeatureSize, getLabels, getLabelSize, getRowDirect, getRowFeatures, matchingTranslatorOptions, prepareFeaturizers

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

getData, getData, getData, getData, newSubDataset, newSubDataset, randomSplit, size, subDataset, subDataset, subDataset, subDataset, toArray

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

prepare

Field Detail

csvUrl

protected java.net.URL csvUrl

csvFormat

protected org.apache.commons.csv.CSVFormat csvFormat

csvRecords

protected java.util.List<org.apache.commons.csv.CSVRecord> csvRecords

Constructor Detail

CsvDataset

protected CsvDataset(CsvDataset.CsvBuilder<?> builder)

Method Detail

getCell

public java.lang.String getCell(long rowIndex,
                                java.lang.String featureName)

Returns a cell in the dataset.

Specified by:

getCell in class TabularDataset

Parameters:

rowIndex - the row index or record index for the cell

featureName - the feature or column of the cell

Returns:

the value of the cell at that row and column

availableSize

protected long availableSize()

Specified by:

availableSize in class ai.djl.training.dataset.RandomAccessDataset

prepare

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

Throws:

java.io.IOException

builder

public static CsvDataset.CsvBuilder<?> builder()

Creates a builder to build a AmesRandomAccess.

Returns:

a new builder

getColumnNames

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

Returns the column names of the CSV file.

Returns:

a list of column name