ai.djl.basicdataset.tabular

Class AmesRandomAccess

java.lang.Object

ai.djl.training.dataset.RandomAccessDataset

ai.djl.basicdataset.tabular.CsvDataset

ai.djl.basicdataset.tabular.AmesRandomAccess

All Implemented Interfaces:

ai.djl.training.dataset.Dataset

public class AmesRandomAccess
extends CsvDataset

Ames house pricing dataset from https://www.kaggle.com/c/house-prices-advanced-regression-techniques/data.

80 features

Training Set: 1460 Records

Test Set: 1459 Records

Can enable/disable features Set one hot vector for categorical variables

Nested Class Summary

Nested Classes

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

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

CsvDataset.CsvBuilder<T extends CsvDataset.CsvBuilder<T>>, CsvDataset.Feature, CsvDataset.Featurizer

Nested classes/interfaces inherited from class ai.djl.training.dataset.RandomAccessDataset

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

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

ai.djl.training.dataset.Dataset.Usage

Field Summary

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

csvFormat, csvRecords, csvUrl, features, labels

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

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

Method Summary

Modifier and Type	Method and Description
static AmesRandomAccess.Builder	builder() Creates a builder to build a AmesRandomAccess.
void	prepare(ai.djl.util.Progress progress)

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

availableSize, get, getColumnNames, toNDList

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

getData, getData, getData, getData, randomSplit, size, 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

Method Detail

prepare

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

Specified by:

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

Overrides:

prepare in class CsvDataset

Throws:

java.io.IOException

builder

public static AmesRandomAccess.Builder builder()

Creates a builder to build a AmesRandomAccess.

Returns:

a new builder