Generalization on Text-based Games using Structured Belief Representations