image

Type Members

1. case class CIFARDataset(datasetType: DatasetType, trainImages: Tensor, trainLabels: Tensor, testImages: Tensor, testLabels: Tensor) extends Product with Serializable

   

2. case class MNISTDataset(datasetType: DatasetType, trainImages: Tensor, trainLabels: Tensor, testImages: Tensor, testLabels: Tensor) extends Product with Serializable

   

3. case class STL10Dataset(trainImages: Tensor, trainLabels: Tensor, testImages: Tensor, testLabels: Tensor, unlabeledImages: Tensor) extends Product with Serializable