Rework FeatureCalculator API to return feature transform matrix instead of features directly