54 lines
1.8 KiB
Python
54 lines
1.8 KiB
Python
"""Install ProtoTorch."""
|
|
|
|
from setuptools import setup
|
|
from setuptools import find_packages
|
|
|
|
PROJECT_URL = 'https://github.com/si-cim/prototorch'
|
|
DOWNLOAD_URL = 'https://github.com/si-cim/prototorch.git'
|
|
|
|
with open('README.md', 'r') as fh:
|
|
long_description = fh.read()
|
|
|
|
setup(name='prototorch',
|
|
version='0.0.0-dev1',
|
|
description='Highly extensible, GPU-supported '
|
|
'Learning Vector Quantization (LVQ) toolbox '
|
|
'built using PyTorch and its nn API.',
|
|
long_description=long_description,
|
|
long_description_content_type='text/markdown',
|
|
author='Jensun Ravichandran',
|
|
author_email='jjensun@gmail.com',
|
|
url=PROJECT_URL,
|
|
download_url=DOWNLOAD_URL,
|
|
license='MIT',
|
|
install_requires=[
|
|
'torch>=1.3.1',
|
|
'torchvision>=0.5.0',
|
|
'numpy>=1.9.1',
|
|
'matplotlib',
|
|
'sklearn',
|
|
],
|
|
extras_require={
|
|
'other': [
|
|
'xlrd',
|
|
'pandas',
|
|
'seaborn',
|
|
'imageio',
|
|
],
|
|
'tests': ['pytest'],
|
|
},
|
|
classifiers=[
|
|
'Development Status :: 2 - Pre-Alpha', 'Environment :: Console',
|
|
'Intended Audience :: Developers', 'Intended Audience :: Education',
|
|
'Intended Audience :: Science/Research',
|
|
'License :: OSI Approved :: MIT License',
|
|
'Programming Language :: Python :: 3.6',
|
|
'Programming Language :: Python :: 3.7',
|
|
'Programming Language :: Python :: 3.8',
|
|
'Operating System :: OS Independent',
|
|
'Topic :: Scientific/Engineering :: Artificial Intelligence',
|
|
'Topic :: Software Development :: Libraries',
|
|
'Topic :: Software Development :: Libraries :: Python Modules'
|
|
],
|
|
packages=find_packages())
|