Refactoring of batch generator class
I have created a new method in the BatchGenerator class called get_samples() which takes the sample indices as input argument and returns the corresponding X,Y. This method can be called manually by the user, but is also called implicitly by the next() method.