Disentangled behavioural representations