large train set, small test set