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))