A simple class for handling data sets,

tefla.dataset.base.Dataset (name, decoder, data_dir=None, num_classes=10, num_examples_per_epoch=1, batch_size=1, items_to_descriptions=None, **kwargs)

Args

  • name: a string, Name of the class instance
  • decoder: object instance, tfrecords object decoding and image encoding and decoding
  • data_dir: a string, path to the data folder
  • num_classes: num of classes of the dataset
  • num_examples_per_epoch: total number of examples per epoch
  • items_to_description: a string descriving the items of the dataset

Methods

data_files (self)

Returns

python list of all (sharded) data set files.