fixed the issue where the classifier only produced one label