backed down on size of training set, was too big to complete