search

rkadlec / asreader

This is an implementation of the Attention Sum Reader model as presented in "Text Comprehension with the Attention Sum Reader Network" available at http://arxiv.org/abs/1603.01547.
Other
95 stars 32 forks source link

Visualization question #3

Closed ganganFighting closed 8 years ago

ganganFighting commented 8 years ago

I run the ./quick-start-cnn.sh script, it can output predictions and logs, but there is no html file that visualize the attention. how to set the argument? Thanks a lot

rkadlec commented 8 years ago

Hi, simply erase --no_html from this line https://github.com/rkadlec/asreader/blob/master/asreader/quick-start-generic.sh#L16 this will output attentions every time the model is evaluated on test and valid data. However, this might consume a lot of disk space.

Another option is to train with --no_html and visualize just attention from trained model loaded by --load_model option.