Init commit

1 files changed, 58 insertions, 0 deletions

diff --git a/setup.py b/setup.py
new file mode 100644
index 0000000..e0af815
--- /dev/null
+++ b/setup.py
@@ -0,0 +1,58 @@
+import setuptools
+
+test_packages = [
+    "pytest>=5.4.3",
+    "pytest-cov>=2.6.1"
+]
+
+base_packages = [
+    "sentence-transformers>=0.3.8",
+    "scikit-learn>=0.22.2",
+    "numpy>=1.18.5",
+]
+
+docs_packages = [
+    "mkdocs==1.1",
+    "mkdocs-material==4.6.3",
+    "mkdocstrings==0.8.0",
+]
+
+dev_packages = docs_packages + test_packages
+
+with open("README.md", "r") as fh:
+    long_description = fh.read()
+
+setuptools.setup(
+    name="keybert",
+    packages=["keybert"],
+    version="0.0.1",
+    author="Maarten Grootendorst",
+    author_email="maartengrootendorst@gmail.com",
+    description="KeyBERT performs keyword extraction with state-of-the-art transformer models.",
+    long_description=long_description,
+    long_description_content_type="text/markdown",
+    url="https://github.com/MaartenGr/keyBERT",
+    keywords="nlp bert keyword extraction embeddings",
+    classifiers=[
+        "Programming Language :: Python",
+        "Intended Audience :: Science/Research",
+        "Intended Audience :: Developers",
+        "Topic :: Scientific/Engineering :: Artificial Intelligence",
+        "License :: OSI Approved :: MIT License",
+        "Topic :: Scientific/Engineering",
+        "Operating System :: Microsoft :: Windows",
+        "Operating System :: POSIX",
+        "Operating System :: Unix",
+        "Operating System :: MacOS",
+        "Programming Language :: Python :: 3.7",
+        "Programming Language :: Python :: 3.6",
+        "Programming Language :: Python :: 3.8",
+    ],
+    install_requires=base_packages,
+    extras_require={
+        "test": test_packages,
+        "docs": docs_packages,
+        "dev": dev_packages,
+    },
+    python_requires='>=3.6',
+)
\ No newline at end of file