🎉 快速上手

欢迎使用 BasicTS 教程！本指南将带您逐步完成使用 BasicTS 训练和评估模型的过程。

在深入之前，我们先简单介绍一下 BasicTS。

什么是 BasicTS？

谁应该使用 BasicTS？

核心功能

现在，让我们开始探索如何通过 BasicTS 实现您的时间序列预测项目吧！

⏬ 克隆仓库

首先，克隆 BasicTS 仓库：

cd /path/to/your/project
git clone https://github.com/zezhishao/BasicTS.git

💿 安装依赖项

操作系统

我们建议在 Linux 系统（如 Ubuntu 或 CentOS）上使用 BasicTS。

Python

需要 Python 3.6 或更高版本（建议使用 3.8 或更高版本）。

我们推荐使用 Miniconda 或 Anaconda 来创建虚拟 Python 环境。

PyTorch

BasicTS 对 PyTorch 版本非常灵活。您可以根据 Python 版本安装 PyTorch。我们建议使用 pip 进行安装。

其他依赖项

确保 PyTorch 正确安装后，您可以安装其他依赖项：

pip install -r requirements.txt

示例设置

示例 1：Python 3.11 + PyTorch 2.5.1 + CUDA 12.4 (推荐)

# 安装 Python
conda create -n BasicTS python=3.11
conda activate BasicTS
# 安装 PyTorch
pip install torch==2.5.1 torchvision==0.20.1 torchaudio==2.5.1 --index-url https://download.pytorch.org/whl/cu124
# 安装其他依赖项
pip install -r requirements.txt

示例 2：Python 3.9 + PyTorch 1.10.0 + CUDA 11.1

# 安装 Python
conda create -n BasicTS python=3.9
conda activate BasicTS
# 安装 PyTorch
pip install torch==1.10.0+cu111 torchvision==0.11.0+cu111 torchaudio==0.10.0 -f https://download.pytorch.org/whl/torch_stable.html
# 安装其他依赖项
pip install -r requirements.txt

📦 下载数据集

您可以从 Google Drive 或 百度网盘 下载 all_data.zip 文件。将文件解压到 datasets/ 目录：

cd /path/to/BasicTS # not BasicTS/basicts
unzip /path/to/all_data.zip -d datasets/

这些数据集已预处理完毕，可以直接使用。

🎯 快速教程：三步训练并评估您的模型

第一步：定义您的模型

forward 函数应该遵循 BasicTS 的规范。多层感知机（MLP）模型的示例可以在 examples/arch.py 中找到。

第二步：定义您的执行器

BasicTS 提供了一个统一的标准化流程，位于 basicts.runner.BaseTimeSeriesForecastingRunner。您仍然需要在 执行器 中的 forward 函数中定义具体的前向过程。

幸运的是，BasicTS 已提供了一个可直接使用的实现，basicts.runner.SimpleTimeSeriesForecastingRunner，可处理大多数情况。MLP 模型的执行器可以使用这个内置执行器。

第三步：配置您的配置文件

所有流程细节和超参数都可以在 .py 文件中配置。该配置文件允许您导入模型和执行器，并设置所有选项，如模型、执行器、数据集、数据缩放器、优化器、损失函数和其他超参数。MLP 模型在 PEMS08 数据集上的配置示例可在 examples/regular_config.py 中找到。

🥳 运行它！

basicts.launch_training 是训练的入口点。您可以运行以下命令来训练您的模型：

• 训练上述提到的 MLP 模型

  python experiments/train.py -c examples/regular_config.py -g 0
  

或者：

• 复现其他内置模型

  BasicTS 提供了多种内置模型。您可以通过以下命令复现这些模型：

  python experiments/train.py -c baselines/${MODEL_NAME}/${DATASET_NAME}.py --gpus '0'
  

  将 ${DATASET_NAME} 和 ${MODEL_NAME} 替换为任意支持的数据集和模型。例如，在 METR-LA 数据集上运行 Graph WaveNet 模型：

  python experiments/train.py -c baselines/GWNet/METR-LA.py --gpus '0'
  

如何评估您的模型

basicts.launch_evaluation 是评估的入口点。您可以运行以下命令来评估您的模型：

python experiments/evaluate.py -cfg {CONFIG_FILE}.py -ckpt {CHECKPOINT_PATH}.pth -g 0

🧑‍💻 进一步探索

本教程为您提供了 BasicTS 的基础知识，但还有更多内容等待您探索。在深入其他主题之前，我们先更详细地了解 BasicTS 的结构：

BasicTS 的核心组件包括 Dataset、Scaler、Model、Metrics、Runner 和 Config。为简化调试过程，BasicTS 作为一个本地化框架运行，所有代码都直接在您的机器上运行。无需 pip install basicts，只需克隆仓库，即可本地运行代码。

以下是一些高级主题和附加功能，帮助您充分利用 BasicTS：

• 🎉 快速上手
• 💡 了解 BasicTS 的设计理念
• 📦 探索数据集设计并自定义数据集
• 🛠️ 了解数据缩放器设计并创建自定义缩放器
• 🧠 深入了解模型设计并构建自定义模型
• 📉 了解评估指标设计并开发自定义损失函数与评估指标
• 🏃‍♂️ 掌握执行器设计并创建自定义执行器
• 📜 解析配置文件设计并自定义配置
• 🎯 探索使用BasicTS进行时间序列分类
• 🔍 探索多种基线模型