Temporally finish the feature extraction, should go back to training data