Gotcha, I think that makes sense. And also in Spark, you can have caching/persist as well to store intermediate results. So I think in the future we can generalize it to the FeatureBuilder class as well :)
Originally posted by @HaoXuAI in #5526 (comment)