mirror of
https://github.com/wassname/pytorch-ts.git
synced 2026-07-18 12:40:51 +08:00
Kashif RasulandGitHub
0053eceaf1
* initial tft model * fixes * fix FeatureProjector * fixed attention module * added quantile loss output * added predict * added tft_transform from gluonts * requires tensor_split * comment out key_padding_mask give nans on GPU * added example
32 lines
903 B
Python
32 lines
903 B
Python
from setuptools import setup, find_packages
|
|
|
|
setup(
|
|
name="pytorchts",
|
|
version="0.3.1",
|
|
description="PyTorch Probabilistic Time Series Modeling framework",
|
|
long_description=open("README.md").read(),
|
|
long_description_content_type="text/markdown",
|
|
author="Kashif Rasul",
|
|
author_email="kashif.rasul@zalando.de",
|
|
url="https://github.com/zalandoresearch/pytorch-ts",
|
|
license="MIT",
|
|
packages=find_packages(exclude=["tests"]),
|
|
include_package_data=True,
|
|
zip_safe=True,
|
|
python_requires=">=3.6",
|
|
install_requires=[
|
|
"torch>=1.8.0",
|
|
"gluonts @ git+https://github.com/awslabs/gluon-ts.git@master#egg=gluonts",
|
|
"holidays",
|
|
"numpy~=1.16",
|
|
"pandas~=1.1",
|
|
"scipy",
|
|
"tqdm",
|
|
"matplotlib",
|
|
"tensorboard",
|
|
"wandb",
|
|
],
|
|
test_suite="tests",
|
|
tests_require=["flake8", "pytest"],
|
|
)
|