Merge branch 'master' of github.com:gchrupala/visually-grounded-speech

gchrupala · Jun 9, 2017 · 4b22735 · 4b22735
2 parents c83a9f4 + f662217
commit 4b22735
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 1 deletion.
diff --git a/README.md b/README.md
@@ -16,6 +16,15 @@ The files in `models.tgz` contain the pre-trained models used for the analyses i
 
 After unpacking these files you should have the directories `data` and `models`.
 
+
+For the scripts in [analysis](analysis), you also need to install:
+
+- sklearn
+- pandas
+- matplotlib
+- keras
+
+
 ## Usage
 
 ### Training models

diff --git a/analysis/README.md b/analysis/README.md
@@ -41,7 +41,7 @@ The plot will be written to `sentlength.pdf`.
 
 In order to generate the figure:
 ```
-python2.7 predict-word-presence.py
+KERAS_BACKEND=theano python2.7 predict-word-presence.py
 ```
 The plot will be written to `predword.pdf`.