(1) Moved to FeatureGenerator
(2) Pickle to save model
(3) Limit memory