add functions for set train and set eval in original traj pred

add functions for set train and set eval in original traj pred #45