modefy line 49, XY4_test_train ->XY5_test_train