安装

使用pip或conda安装NeuralForecast

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

pip install neuralforecast

或者

conda install -c conda-forge neuralforecast
Tip

神经预测方法可以利用 GPU 计算。确保已安装 Cuda。

Warning

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

Tip

我们建议您在 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]"

Give us a ⭐ on Github