Template Class DatasetBase

• Defined in File dataset.h

Class Documentation

template<class SampleType, class Iterator, class Batch>
class DatasetBase

Public Types

typedef Batch batch_type

typedef Ptr<Batch> batch_ptr

typedef Iterator iterator

typedef SampleType Sample

Public Functions

DatasetBase(std::vector<std::string> paths, Ptr<Options> options)

DatasetBase(Ptr<Options> options)

virtual Iterator begin() = 0

virtual Iterator end() = 0

virtual void shuffle() = 0

virtual Sample next() = 0

virtual batch_ptr toBatch(const std::vector<Sample>&) = 0

virtual void reset()

virtual void prepare()

virtual void restore(Ptr<TrainingState>)

virtual Ptr<Options> options()

Protected Attributes

std::vector<std::string> paths_

Ptr<Options> options_

bool inference_ = {false}