Výměna parametru TEST za parametr TRAIN.