trainingset fixed

Signed-off-by: shyeon.kim <shyeon.kim@scipi.net>