What is the best way to read SQL dataset in to Tensorflow?
You use the methods on SqlDataset to manipulate the data. For example, create a train/test split with:

test_dataset = dataset.take(1000) 
train_dataset = dataset.skip(1000)


I would get rid of unneeded columns in the SELECT statement to reduce the size of the data as early as possible.