Semi-supervised K-Means clustering for trajectory analysis in behavioural experiments