Generating the training datasets