jimfleming / recurrent-entity-networks

TensorFlow implementation of "Tracking the World State with Recurrent Entity Networks".
https://arxiv.org/abs/1612.03969
MIT License
273 stars 69 forks source link

Training EntNet on CBT dataset #20

Closed JingHanZhan closed 2 years ago

JingHanZhan commented 5 years ago

I have read the paper "TRACKING THE WORLD STATE WITH RECURRENT ENTITY NETWORKS". In the section 5 experiments, table 4 shows the result of testing Entnet on CBT(children's book test). I have some problem about training it. In CBT dataset it looks like {story, query, candidate, answer}. And bAbI is {story, query, answer} If I want to train CBT dataset, how can I feed candidate to the model ? Or candidate only used to prepare data as window sentence ?

Thanks!