Skip to content
Snippets Groups Projects
Commit 5bb1f547 authored by Tim O'Donnell's avatar Tim O'Donnell
Browse files

Pin keras version

Pin keras version. Also add note on KERAS_BACKEND to README.md.

Closes #67. Closes #65.
parent eeb07c9a
No related merge requests found
......@@ -13,7 +13,16 @@ The MHCflurry predictors are implemented in Python using [keras](https://keras.i
## Setup
Install the package:
To configure keras, the neural network library used by MHCflurry, you'll need to set an environment variable in your shell:
```
export KERAS_BACKEND=theano
```
If you're familiar with keras, you may also try using the tensorflow backend. MHCflurry is currently tested using theano, however.
Now install the package:
```
pip install mhcflurry
......
......@@ -2,7 +2,7 @@ numpy>= 1.11
pandas>=0.13.1
appdirs
theano>=0.8.2
keras
keras==1.1.0
fancyimpute>=0.0.12
scikit-learn
h5py
......
......@@ -76,7 +76,7 @@ if __name__ == '__main__':
'pandas>=0.13.1',
'appdirs',
'theano>=0.8.2',
'keras',
'keras==1.1.0',
'fancyimpute>=0.0.12',
'scikit-learn',
'h5py',
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment