def create dataset(xs,ys,n_classes=10): ys=tf.one_hot(ys,depth=n_classes) return tf.data.Dataset.from_tensor_slices(xs,ys).map(preprocess).shuffle(len(ys)).batch(128)