https://mingredbestcheapprice.blogspot.com/2024/03/tensorflow-keras-with-tf-dataset-input.html