diff --git a/.gitignore b/.gitignore index 33301d1..8220d82 100644 --- a/.gitignore +++ b/.gitignore @@ -1,14 +1,36 @@ +# Python build *.pyc + +# Tensorflow files *.json + +# Pycharm Project .idea/* +.idea + +# Virtualenv .venv/* -boltzmann_machines/test_rbm_1/* -boltzmann_machines/test_rbm_2/* + +models/* +!models/fetch_models.sh data/*.npy data/mnist/ data/cifar-10-batches-py/ + +# Notebooks notebooks/.ipynb_checkpoints/* -models/* -!models/fetch_models.sh + +# LaTex files tex/*.swp -.idea + +# python setup.py develop +# python setup_cpu.py develop +boltzmann_machines.egg-info/ + +# Coverage +.coverage + +# Research projects +.tests/ +boltzmann_machines/test_rbm_1/* +boltzmann_machines/test_rbm_2/* \ No newline at end of file diff --git a/setup.py b/setup.py index 521b2a8..631d186 100644 --- a/setup.py +++ b/setup.py @@ -4,7 +4,7 @@ setup(name='boltzmann_machines', version='0.1', url='https://github.com/monsta-hd/boltzmann-machines', - install_requires=[ 'tensorflow-gpu>=1.3.0', + install_requires=['tensorflow-gpu>=1.3.0', 'scipy>=0.17', 'keras>=2.0.8', 'matplotlib>=1.5.3', @@ -13,6 +13,7 @@ 'numpy>=1.13.0', 'scikit-learn>=0.19.0', 'seaborn>=0.8.1', - 'tqdm>=4.14.0' + 'tqdm>=4.14.0', + 'coverage>=4.5.0' ], packages=find_packages()) diff --git a/setup_cpu.py b/setup_cpu.py new file mode 100644 index 0000000..a2608b9 --- /dev/null +++ b/setup_cpu.py @@ -0,0 +1,19 @@ +from setuptools import setup +from setuptools import find_packages + +setup(name='boltzmann_machines', + version='0.1', + url='https://github.com/monsta-hd/boltzmann-machines', + install_requires=[ 'tensorflow>=1.13.0', + 'scipy>=0.17', + 'keras>=2.0.8', + 'matplotlib>=1.5.3', + 'nose-exclude>=0.5.0', + 'nose>=1.3.4', + 'numpy>=1.13.0', + 'scikit-learn>=0.19.0', + 'seaborn>=0.8.1', + 'tqdm>=4.14.0', + 'coverage>=4.5.0' +], +packages=find_packages())