def __init__(self, batch_size):
Dataset.__init__(self, batch_size)
in_reshape = (-1, 28, 28, 1)
self.set_train(
DataValues(x=mnist.train.images.reshape(*in_reshape),
y=mnist.train.labels))
self.set_val(
DataValues(x=mnist.test.images.reshape(*in_reshape),
y=mnist.test.labels))
self.set_test(
DataValues(x=mnist.test.images.reshape(*in_reshape),
y=mnist.test.labels))
def __init__(self, batch_size):
Dataset.__init__(self, batch_size)
in_reshape = (-1, N_STEPS, N_STEPS)
self.set_train(
DataValues(x=mnist.train.images.reshape(*in_reshape),
y=mnist.train.labels))
print("Label size is ", self.get_train().y.shape)
self.set_val(
DataValues(x=mnist.test.images.reshape(*in_reshape),
y=mnist.test.labels))
self.set_test(
DataValues(x=mnist.test.images.reshape(*in_reshape),
y=mnist.test.labels))
def __init__(self, batch_size):
Dataset.__init__(self, batch_size)
self.set_train(DataValues(x=mnist.train.images, y=mnist.train.labels))
self.set_val(DataValues(x=mnist.test.images, y=mnist.test.labels))
self.set_test(DataValues(x=mnist.test.images, y=mnist.test.labels))