Telling visualize.py to load a test set too.