InputLabels

is an option for PrepareTrainingData. Can be set to a list of integers corresponding to the labels to be used from the given segmentation.