Support testing on training data