Implementation of training data set generator