/TS-Lib

来源于Time-Series-Library,对其中的训练逻辑进行了修改

Primary LanguagePythonMIT LicenseMIT

Time Series Library (TSlib)

时间序列仓库,用于单变量或多变量时间序列的预训练分类回归异常检测预测任务

Time Series Library的基础上进行了修改,将每个任务模块进行剥离,让骨干网络和头部模块化,方便做工程项目

TODO

  • 多GPU支持

  • tensorboard可视化

  • 各类任务数据集的探索性数据分析(exploratory data analysis)

  • 对训练代码进行封装

Tasks

1. Pretrain

分离骨干网络,对骨干网络做预训练

  • self-supervised pretrain
  • masked reconstruction pretrain
  • next series prediction pretrain

2. Classification

单变量时序分类和多变量时序分类

3. Regression

TODO

4. Prediction

TODO

5. Imputation

TODO

Usage

TODO