mistake in main.py

[Voyager0057]

"test_iter" is not defined

[idealer-liXi]

the MR dataset was no standard train/test split, and thus used the 10-fold CV