dataset aggregation in Prune Model.