On the Generalisation of Koopman Representations for Chaotic System Control