removed total training set test due to length of execution