您可以从 Python 包索引 使用以下命令安装 NeuralForecast发布版本

pip install neuralforecast

或者

conda install -c conda-forge neuralforecast

提示

神经预测方法受益于使用 GPU 计算。请确保已安装 Cuda。

警告

我们正在不断更新 neuralforecast,建议固定版本以避免问题。pip install neuralforecast=="1.0.0"

提示

我们建议将库安装在 Python 虚拟环境或 conda 环境 中。

附加功能

您可以使用以下附加功能添加可选功能

  • 使用 spark 进行分布式训练:pip install neuralforecast[spark]
  • 从 S3 保存和加载:pip install neuralforecast[aws]

使用我们的环境(可选)

如果您没有 Conda 环境且需要 Numba、Pandas、NumPy、Jupyter、Tune 和 Nbdev 等工具,可以按照以下步骤使用我们的环境:

  1. 克隆 NeuralForecast 仓库
$ git clone https://github.com/Nixtla/neuralforecast.git && cd neuralforecast
  1. 使用 environment.yml 文件创建环境
$ conda env create -f environment.yml
  1. 激活环境
$ conda activate neuralforecast
  1. 安装 NeuralForecast 开发版本
$ pip install -e ".[dev]"