add comparison of training size