Don't load the entire dataset into memory.
Use a generator function instead.