predictAssignments function

Predict the cluster assignments for new trajectories