Generating training dataset