How to create a tensorflow function?
def dataframe_to_tf(X,y,shuffle=True): def input_function(): dataset=tf.data.Dataset.from_tensor_slices((dict(X),y)) dataset=dataset.shuffle(1000) dataset=dataset.batch(32).repeat(10) return dataset return input_function