Learning Belief Representations for Imitation Learning in POMDPs