https://lacucinadiadine.blogspot.com/2024/04/how-to-write-generator-for-keras-model.html