cifar10

README

CIFAR-10 Dataset

The dataset itself is not included in the repository. Run download.sh in the data subdirectory to download the data.

Documentation

Overview

Package cifar10 provides an interface to read the cifar-10 dataset.

Index

• type DataSet
  • func (d *DataSet) HasNext() bool
  • func (d *DataSet) Next() (image []byte, label byte)
• type DataSource
  • func NewTestDataSource(to tensor.Operator) *DataSource
  • func NewTrainingDataSource(to tensor.Operator) *DataSource
  • func (ds *DataSource) NextBatch(batchSize int) (data tensor.Tensor, label []int)
  • func (ds *DataSource) Rewind()

Types

type DataSet

type DataSet struct {
	// contains filtered or unexported fields
}

The DataSet can provide Cifar-10 data.

func (*DataSet) HasNext

func (d *DataSet) HasNext() bool

HasNext returns true if there are more images in the dataset.

func (*DataSet) Next

func (d *DataSet) Next() (image []byte, label byte)

Next returns the next image and label. It panics if there is no more images in the dataset.

type DataSource

type DataSource struct {
	// contains filtered or unexported fields
}

A DataSource provides convenient solution to feed data to neural networks.

func NewTestDataSource

func NewTestDataSource(to tensor.Operator) *DataSource

NewTestDataSource creates a DataSource object that fetches data from the test set.

func NewTrainingDataSource

func NewTrainingDataSource(to tensor.Operator) *DataSource

NewTrainingDataSource returns a DataSource object that fetches data from the training set.

func (*DataSource) NextBatch

func (ds *DataSource) NextBatch(batchSize int) (
	data tensor.Tensor,
	label []int,
)

NextBatch returns another batch of data.

func (*DataSource) Rewind

func (ds *DataSource) Rewind()

Rewind resets the pointer to the beginning of the dataset.