def make_dataset(images,labels,epochs,batch_size,shuffle=True):
dataset = tf.data.Dataset.from_tensor_slices((images,labels))
if shuffle:
dataset = dataset.shuffle(10000)
dataset = dataset.repeat(epochs).batch_size(batch_size).prefecth(50)
return dataset