diff --git a/setup.py b/setup.py index 9f84595..e43773e 100644 --- a/setup.py +++ b/setup.py @@ -2,7 +2,7 @@ install_requires = ['scipy==0.19.1', 'numpy==1.15.2', - 'tensorflow==1.5.0', + 'tensorflow==2.5.0', 'scikit_learn==0.19.1', 'Keras==2.0.6', 'h5py==2.7.1']