CIFAR-10 Data Set¶
-
class
deepobs.tensorflow.datasets.cifar10.
cifar10
(batch_size, data_augmentation=True, train_eval_size=10000)[source]¶ DeepOBS data set class for the CIFAR-10 data set.
Parameters: - batch_size (int) -- The mini-batch size to use. Note that, if
batch_size
is not a divider of the dataset size (50 000
for train,10 000
for test) the remainder is dropped in each epoch (after shuffling). - data_augmentation (bool) -- If
True
some data augmentation operations (random crop window, horizontal flipping, lighting augmentation) are applied to the training data (but not the test data). - train_eval_size (int) -- Size of the train eval data set.
Defaults to
10 000
the size of the test set.
-
batch
¶ A tuple
(x, y)
of tensors, yielding batches of CIFAR-10 images (x
with shape(batch_size, 32, 32, 3)
) and corresponding one-hot label vectors (y
with shape(batch_size, 10)
). Executing these tensors raises atf.errors.OutOfRangeError
after one epoch.
-
train_init_op
¶ A tensorflow operation initializing the dataset for the training phase.
-
train_eval_init_op
¶ A tensorflow operation initializing the testproblem for evaluating on training data.
-
test_init_op
¶ A tensorflow operation initializing the testproblem for evaluating on test data.
-
phase
¶ A string-value tf.Variable that is set to
train
,train_eval
ortest
, depending on the current phase. This can be used by testproblems to adapt their behavior to this phase.
- batch_size (int) -- The mini-batch size to use. Note that, if