Move split into training/testing into FeatureExtractor