Adding training sample counting for individual trees.