This repository has been archived by the owner on Jan 5, 2023. It is now read-only.
-
Notifications
You must be signed in to change notification settings - Fork 32
/
setup.py
52 lines (48 loc) · 1.93 KB
/
setup.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
# -*- coding: utf-8 -*-
import os
import sys
from setuptools import setup
import nmtpy
# Install pycocoevalcap metric scorers as well
pycocometrics = ['bleu', 'meteor', 'cider', 'rouge']
pycocopackages = ['nmtpy.cocoeval.%s' % m for m in pycocometrics]
if 'install' in sys.argv or 'develop' in sys.argv:
if not os.path.exists('nmtpy/external/data/paraphrase-en.gz'):
print('You need to run scripts/get-meteor-data.sh first.')
sys.exit(1)
setup(
name='nmtpy',
version=nmtpy.__version__,
description='Neural Machine Translation Framework in Python',
url='https://github.com/lium-lst/nmtpy',
author='Ozan Çağlayan',
author_email='[email protected]',
license='MIT',
classifiers=[
'Development Status :: 5 - Production/Stable',
'Intended Audience :: Science/Research',
'Topic :: Scientific/Engineering',
'License :: OSI Approved :: MIT License',
'Programming Language :: Python :: 3 :: Only',
'Programming Language :: Python :: 3.5',
'Operating System :: POSIX',
],
keywords='nmt neural-mt translation deep-learning',
packages=['nmtpy', 'nmtpy.models', 'nmtpy.iterators', 'nmtpy.metrics', 'nmtpy.cocoeval'] + pycocopackages,
package_data={'' : ['external/meteor-1.5.jar', 'external/data/*gz', 'external/multi-bleu.perl']}, # data files
install_requires=[
'numpy',
'theano',
],
scripts=[
'bin/nmt-train',
'bin/nmt-extract',
'bin/nmt-rescore',
'bin/nmt-translate',
'bin/nmt-translate-factors', # Factored NMT variant.
'bin/nmt-build-dict',
'bin/nmt-coco-metrics',
'bin/nmt-bpe-apply',
'bin/nmt-bpe-learn',
],
zip_safe=False)