安装#

GluonTS 可以通过 PyPi 获取:

pip install gluonts

注意

GluonTS 使用 语义化版本控制 来管理版本。由于该库正在积极开发,我们使用 v0 作为主要版本。

我们计划在每个季度末发布一个新的小版本。当前计划的发布可以在 GitHub 上找到。

版本保证

重大变更仅在新的次要版本中引入。当前版本提供错误修复和小幅改进,并根据需要发布。

对于生产环境的使用,我们建议在安装GluonTS时限制版本:

pip install gluonts==0.12.*

可选和额外依赖#

重要

GluonTS使用最小依赖模型。

这意味着要使用大多数模型和功能,需要安装额外的依赖项。

Python 有 extras 的概念 – 可以选择性安装的依赖,以解锁包的某些功能。

在安装一个包时,它们通过 [...] 传递,位于包名之后,但在版本说明符之前:

pip install "some-package[extra-1,extra-2]==version"

我们在GluonTS中广泛使用可选依赖项,以保持所需依赖项的数量最小。为了仍然允许用户选择某些功能,我们公开了许多额外的依赖项。

例如,我们提供对使用 Apache Arrow 的 Arrow 和 Parquet 基于数据集的读取和写入的支持。然而,这是一项沉重的依赖,尤其是如果一个人不需要它。因此,我们提供 arrow-extra,它安装所需的包并且可以简单地使用以下方式启用:

pip install "gluonts[arrow]"

模型#

PyTorch#

使用 PyTorch 编写的模型可以通过 gluonts.torch 子包获取。

除了PyTorch,我们还需要安装PyTorch Lightning

这两个必要的依赖已包含在 torch-extra中:

pip install "gluonts[torch]"

MXNet#

基于MXNet的模型需要安装一个版本的 mxnet

注意

MXNet提供了针对CPU和GPU使用的不同包。请参考它的 documentation以 选择适合您用例的正确版本。

mxnet-extra将安装一个仅支持CPU的版本:

pip install "gluonts[mxnet]"

第三方#

R-预测#

GluonTS 包括一个薄包装器,用于调用 R forecast 包。

为了使用它,您需要安装 R 和 安装 forecast 包:

R -e 'install.packages(c("forecast", "nnfor"), repos="https://cloud.r-project.org")'

您还需要安装 R 的附加依赖项:

pip install "gluonts[R]"
预言家#

Prophet 预测库可通过 gluonts.model.prophet 获取,并且需要安装 prophet 包。你可以通过安装 prophet-extra 来获取它:

pip install "gluonts[prophet]"

数据集#

JSON#

由于Python内置的 json 包被认为相对较慢,我们使用更快的实现(如果可用): orjson(推荐)和 ujson

您可以通过 orjson 安装:

pip install orjson

提示

如果没有安装 orjsonujson,GluonTS 将发出警告。 不同实现之间没有功能上的差异,但 尤其是在处理较大数据集时,依赖默认的 json 包可能会显著影响性能。

箭头#

GluonTS 支持 Parquet 文件,使用 PyArrow

此外,arrow的自定义数据格式也得到了支持。

要使用这些,您可以安装 pyarrow 包或使用 arrow-extra:

pip install "gluonts[arrow]"

其他#

#

shell 模块提供与 Amazon SageMaker 的集成,并可通过以下方式获取:

pip install "gluonts[shell]"