From 87c1dc2a1a2770d95103bfb51e93897d306f39c5 Mon Sep 17 00:00:00 2001 From: Andreas Leitherer <leitherer@fhi-berlin.mpg.de> Date: Tue, 5 Jan 2021 12:04:43 +0100 Subject: [PATCH] Modified setup.py file for nn regression --- setup.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/setup.py b/setup.py index b6cea0d..264d91d 100644 --- a/setup.py +++ b/setup.py @@ -13,5 +13,5 @@ setup( description=metainfo['title'], long_description=metainfo['description'], packages=find_packages(), - install_requires=['tensorflow==1.14.0', 'keras==2.3.0', 'numpy==1.16.4', 'scipy', 'matplotlib', 'pandas', 'seaborn', 'pymatgen==2020.3.13', 'sklearn'], + install_requires=['tensorflow==1.13.1', 'keras==2.2.4', 'keras-vis', 'numpy==1.16.4', 'scipy==1.3.1', 'matplotlib', 'pandas', 'seaborn', 'pymatgen==2020.3.13', 'sklearn'], ) -- GitLab