TODO: save greedy_ensemble