Class FashionMnist
- java.lang.Object
-
- ai.djl.training.dataset.RandomAccessDataset
-
- ai.djl.training.dataset.ArrayDataset
-
- ai.djl.basicdataset.cv.classification.FashionMnist
-
- All Implemented Interfaces:
ai.djl.training.dataset.Dataset
public final class FashionMnist extends ai.djl.training.dataset.ArrayDatasetFashMnist is a dataset from Zalando article images https://github.com/zalandoresearch/fashion-mnist.Each sample is an image (in 3-D NDArray) with shape (28, 28, 1).
-
-
Nested Class Summary
Nested Classes Modifier and Type Class Description static classFashionMnist.BuilderA builder for aFashionMnist.
-
Field Summary
Fields Modifier and Type Field Description static intIMAGE_HEIGHTstatic intIMAGE_WIDTHstatic intNUM_CLASSES
-
Method Summary
All Methods Static Methods Instance Methods Concrete Methods Modifier and Type Method Description static FashionMnist.Builderbuilder()Creates a builder to build aMnist.voidprepare(ai.djl.util.Progress progress)-
Methods inherited from class ai.djl.training.dataset.RandomAccessDataset
getData, getData, getData, getData, randomSplit, size, subDataset, toArray
-
-
-
-
Field Detail
-
IMAGE_WIDTH
public static final int IMAGE_WIDTH
- See Also:
- Constant Field Values
-
IMAGE_HEIGHT
public static final int IMAGE_HEIGHT
- See Also:
- Constant Field Values
-
NUM_CLASSES
public static final int NUM_CLASSES
- See Also:
- Constant Field Values
-
-
Method Detail
-
builder
public static FashionMnist.Builder builder()
Creates a builder to build aMnist.- Returns:
- a new builder
-
prepare
public void prepare(ai.djl.util.Progress progress) throws java.io.IOException- Specified by:
preparein interfaceai.djl.training.dataset.Dataset- Overrides:
preparein classai.djl.training.dataset.ArrayDataset- Throws:
java.io.IOException
-
-