fixed if the feature selected count is less than defined set of training labels