add training set compuation