diff --git a/requirements.txt b/requirements.txt index f32e7a3c2d3b976a3148ba526617e569d6fa09e0..8fee4cf4e049c18783e59cfb9fcde0982a755470 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,7 +1,7 @@ six numpy>=1.11 pandas>=0.20.3 -Keras==2.0.9 +Keras==keras-2.1.2 tensorflow>=1.1.0 appdirs scikit-learn diff --git a/setup.py b/setup.py index 92af2813f3b4990433a3b70d9c704004ba03f4a9..1d5361634f0fdc9b728d94254e88d6f9653ab55b 100644 --- a/setup.py +++ b/setup.py @@ -51,7 +51,7 @@ if __name__ == '__main__': 'six', 'numpy>=1.11', 'pandas>=0.20.3', - 'Keras==2.0.9', + 'Keras==2.1.2', 'appdirs', 'tensorflow>=1.1.0', 'scikit-learn',