refactoring ModelBuilder, FeatureSet, and TrainingSet