Add a method to extract the feature part of the label