OutputLabels

is an option for PrepareTrainingData. Can be set to a list of integers. The used label number will be replaced by these numbers.