2019-02-12 00:27:09 +08:00
|
|
|
import setuptools
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
with open("README.md", "r") as fh:
|
|
|
|
|
|
|
|
|
|
long_description = fh.read()
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
REQUIRED_PACKAGES = [
|
2022-06-22 02:24:09 +08:00
|
|
|
# 'tensorflow>=1.4.0',
|
|
|
|
|
'gensim>=4.0.0',
|
|
|
|
|
'networkx',
|
|
|
|
|
'joblib',
|
|
|
|
|
'fastdtw',
|
2019-04-22 00:02:08 +08:00
|
|
|
'tqdm',
|
2019-04-16 01:18:57 +08:00
|
|
|
'numpy',
|
2019-04-22 00:02:08 +08:00
|
|
|
'scikit-learn',
|
|
|
|
|
'pandas',
|
|
|
|
|
'matplotlib',
|
2022-06-22 02:24:09 +08:00
|
|
|
'deepctr'
|
2019-02-12 00:27:09 +08:00
|
|
|
]
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
setuptools.setup(
|
|
|
|
|
|
|
|
|
|
name="ge",
|
|
|
|
|
|
|
|
|
|
version="0.0.0",
|
|
|
|
|
|
|
|
|
|
author="Weichen Shen",
|
|
|
|
|
|
2022-06-22 02:24:09 +08:00
|
|
|
author_email="weichenswc@163.com",
|
2019-02-12 00:27:09 +08:00
|
|
|
|
|
|
|
|
url="https://github.com/shenweichen/GraphEmbedding",
|
|
|
|
|
|
|
|
|
|
packages=setuptools.find_packages(exclude=[]),
|
|
|
|
|
|
2022-06-22 02:24:09 +08:00
|
|
|
python_requires='>=3.5', # 3.4.6
|
2019-02-12 00:27:09 +08:00
|
|
|
|
|
|
|
|
install_requires=REQUIRED_PACKAGES,
|
|
|
|
|
|
|
|
|
|
extras_require={
|
|
|
|
|
|
2019-07-21 13:11:24 +08:00
|
|
|
"cpu": ['tensorflow>=1.4.0,!=1.7.*,!=1.8.*'],
|
2019-02-12 00:27:09 +08:00
|
|
|
|
2019-07-21 13:11:24 +08:00
|
|
|
"gpu": ['tensorflow-gpu>=1.4.0,!=1.7.*,!=1.8.*'],
|
2019-02-12 00:27:09 +08:00
|
|
|
|
|
|
|
|
},
|
|
|
|
|
|
|
|
|
|
entry_points={
|
|
|
|
|
|
|
|
|
|
},
|
|
|
|
|
license="MIT license",
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
)
|
