Skip to main content

一个功能强大且可扩展的库,可用于各种时间序列数据挖掘任务

项目描述

PyPI 版本 康达锻造版 PyPI 下载 执照 测试状态 代码覆盖率 ReadTheDocs 状态 粘合剂 乔斯 DOI 窝 推特

矮胖的标志

矮墩墩

STUMPY 是一个功能强大且可扩展的库,可以有效地计算称为矩阵轮廓的东西,可用于各种时间序列数据挖掘任务,例如:

  • 模式/基序(较长时间序列中近似重复的子序列)发现

  • 异常/新奇(不和谐)发现

  • 小形状发现

  • 语义分割

  • 流式传输(在线)数据

  • 快速近似矩阵轮廓

  • 时间序列链(按时间顺序排列的子序列模式集)

  • 用于总结长时间序列的片段

  • 用于选择最佳子序列窗口大小的平移矩阵配置文件

  • 和更多 …

无论您是学者、数据科学家、软件开发人员还是时间序列爱好者,STUMPY 都易于安装,我们的目标是让您更快地获得时间序列见解。有关更多信息,请参阅文档

如何使用 STUMPY

请参阅我们的API 文档以获取可用函数的完整列表,并参阅我们的信息教程以获取更全面的示例用例。下面,您将找到快速演示如何使用 STUMPY 的代码片段。

STUMP的典型用法(一维时间序列数据):

import stumpy
import numpy as np

if __name__ == "__main__":
    your_time_series = np.random.rand(10000)
    window_size = 50  # Approximately, how many data points might be found in a pattern

    matrix_profile = stumpy.stump(your_time_series, m=window_size)

Dask 通过STUMPED分布的一维时间序列数据的分布式使用:

import stumpy
import numpy as np
from dask.distributed import Client

if __name__ == "__main__":
    dask_client = Client()

    your_time_series = np.random.rand(10000)
    window_size = 50  # Approximately, how many data points might be found in a pattern

    matrix_profile = stumpy.stumped(dask_client, your_time_series, m=window_size)

使用GPU-STUMP处理一维时间序列数据的GPU 使用情况:

import stumpy
import numpy as np
from numba import cuda

if __name__ == "__main__":
    your_time_series = np.random.rand(10000)
    window_size = 50  # Approximately, how many data points might be found in a pattern
    all_gpu_devices = [device.id for device in cuda.list_devices()]  # Get a list of all available GPU devices

    matrix_profile = stumpy.gpu_stump(your_time_series, m=window_size, device_id=all_gpu_devices)

带有MSTUMP的多维时间序列数据:

import stumpy
import numpy as np

if __name__ == "__main__":
    your_time_series = np.random.rand(3, 1000)  # Each row represents data from a different dimension while each column represents data from the same dimension
    window_size = 50  # Approximately, how many data points might be found in a pattern

    matrix_profile, matrix_profile_indices = stumpy.mstump(your_time_series, m=window_size)

使用 Dask Distributed MSTUMPED进行分布式多维时间序列数据分析:

import stumpy
import numpy as np
from dask.distributed import Client

if __name__ == "__main__":
    dask_client = Client()

    your_time_series = np.random.rand(3, 1000)   # Each row represents data from a different dimension while each column represents data from the same dimension
    window_size = 50  # Approximately, how many data points might be found in a pattern

    matrix_profile, matrix_profile_indices = stumpy.mstumped(dask_client, your_time_series, m=window_size)

具有锚定时间序列链 (ATSC)的时间序列链:

import stumpy
import numpy as np

if __name__ == "__main__":
    your_time_series = np.random.rand(10000)
    window_size = 50  # Approximately, how many data points might be found in a pattern

    matrix_profile = stumpy.stump(your_time_series, m=window_size)

    left_matrix_profile_index = matrix_profile[:, 2]
    right_matrix_profile_index = matrix_profile[:, 3]
    idx = 10  # Subsequence index for which to retrieve the anchored time series chain for

    anchored_chain = stumpy.atsc(left_matrix_profile_index, right_matrix_profile_index, idx)

    all_chain_set, longest_unanchored_chain = stumpy.allc(left_matrix_profile_index, right_matrix_profile_index)

具有快速低成本单能语义分割(FLUSS)的语义分割

import stumpy
import numpy as np

if __name__ == "__main__":
    your_time_series = np.random.rand(10000)
    window_size = 50  # Approximately, how many data points might be found in a pattern

    matrix_profile = stumpy.stump(your_time_series, m=window_size)

    subseq_len = 50
    correct_arc_curve, regime_locations = stumpy.fluss(matrix_profile[:, 1],
                                                    L=subseq_len,
                                                    n_regimes=2,
                                                    excl_factor=1
                                                    )

依赖项

支持的 Python 和 NumPy 版本根据NEP 29 deprecation policy确定。

在哪里得到它

康达安装(首选):

conda install -c conda-forge stumpy

PyPI 安装,假设您安装了 numpy、scipy 和 numba:

python -m pip install stumpy

要从源代码安装 stumpy,请参阅文档中的说明

文档

为了充分理解和欣赏底层算法和应用程序,您必须阅读原始出版物。有关如何使用 STUMPY 的更详细示例,请查阅最新文档或浏览我们的动手教程

表现

我们使用 Numba JIT 编译版本的代码在具有不同长度(即np.random.rand(n))以及不同CPU 和 GPU 硬件资源的随机生成的时间序列数据上测试了计算精确矩阵轮廓的性能。

STUMPY 性能图

原始结果在下表中显示为 Hours:Minutes:Seconds.Milliseconds 并且具有恒定的窗口大小m = 50。请注意,这些报告的运行时间包括将数据从主机移动到所有 GPU 设备所需的时间。您可能需要滚动到表格的右侧才能查看所有运行时。

一世

n = 2

GPU-STOMP

STUMP.2

STUMP.16

难倒.128

难倒.256

GPU-STUMP.1

GPU-STUMP.2

GPU-STUMP.DGX1

GPU-STUMP.DGX2

6

64

00:00:10.00

00:00:00.00

00:00:00.00

00:00:05.77

00:00:06.08

00:00:00.03

00:00:01.63

7

128

00:00:10.00

00:00:00.00

00:00:00.00

00:00:05.93

00:00:07.29

00:00:00.04

00:00:01.66

8

256

00:00:10.00

00:00:00.00

00:00:00.01

00:00:05.95

00:00:07.59

00:00:00.08

00:00:01.69

00:00:06.68

00:00:25.68

9

512

00:00:10.00

00:00:00.00

00:00:00.02

00:00:05.97

00:00:07.47

00:00:00.13

00:00:01.66

00:00:06.59

00:00:27.66

10

1024

00:00:10.00

00:00:00.02

00:00:00.04

00:00:05.69

00:00:07.64

00:00:00.24

00:00:01.72

00:00:06.70

00:00:30.49

11

2048

00:00:00.05

00:00:00.09

00:00:05.60

00:00:07.83

00:00:00.53

00:00:01.88

00:00:06.87

00:00:31.09

12

4096

00:00:00.22

00:00:00.19

00:00:06.26

00:00:07.90

00:00:01.04

00:00:02.19

00:00:06.91

00:00:33.93

13

8192

00:00:00.50

00:00:00.41

00:00:06.29

00:00:07.73

00:00:01.97

00:00:02.49

00:00:06.61

00:00:33.81

14

16384

00:00:01.79

00:00:00.99

00:00:06.24

00:00:08.18

00:00:03.69

00:00:03.29

00:00:07.36

00:00:35.23

15

32768

00:00:06.17

00:00:02.39

00:00:06.48

00:00:08.29

00:00:07.45

00:00:04.93

00:00:07.02

00:00:36.09

16

65536

00:00:22.94

00:00:06.42

00:00:07.33

00:00:09.01

00:00:14.89

00:00:08.12

00:00:08.10

00:00:36.54

17

131072

00:00:10.00

00:01:29.27

00:00:19.52

00:00:09.75

00:00:10.53

00:00:29.97

00:00:15.42

00:00:09.45

00:00:37.33

18

262144

00:00:18.00

00:05:56.50

00:01:08.44

00:00:33.38

00:00:24.07

00:00:59.62

00:00:27.41

00:00:13.18

00:00:39.30

19

524288

00:00:46.00

00:25:34.58

00:03:56.82

00:01:35.27

00:03:43.66

00:01:56.67

00:00:54.05

00:00:19.65

00:00:41.45

20

1048576

00:02:30.00

01:51:13.43

00:19:54.75

00:04:37.15

00:03:01.16

00:05:06.48

00:02:24.73

00:00:32.95

00:00:46.14

21

2097152

00:09:15.00

09:25:47.64

03:05:07.64

00:13:36.51

00:08:47.47

00:20:27.94

00:09:41.43

00:01:06.51

00:01:02.67

22

4194304

36:12:23.74

10:37:51.21

00:55:44.43

00:32:06.70

01:21:12.33

00:38:30.86

00:04:03.26

00:02:23.47

23

8388608

143:16:09.94

38:42:51.42

03:33:30.53

02:00:49.37

05:11:44.45

02:33:14.60

00:15:46.26

00:08:03.76

24

16777216

14:39:11.99

07:13:47.12

20:43:03.80

09:48:43.42

01:00:24.06

00:29:07.84

17729800

09:16:12.00

15:31:31.75

07:18:42.54

23:09:22.43

10:54:08.64

01:07:35.39

00:32:51.55

25

33554432

56:03:46.81

26:27:41.29

83:29:21.06

39:17:43.82

03:59:32.79

01:54:56.52

26

67108864

211:17:37.60

106:40:17.17

328:58:04.68

157:18:30.50

15:42:15.94

07:18:52.91

100000000

291:07:12.00

234:51:35.39

35:03:44.61

16:22:40.81

27

134217728

64:41:55.09

29:13:48.12

硬件资源

GPU-STOMP:这些结果复制自原始的Matrix Profile II论文 - NVIDIA Tesla K80(包含 2 个 GPU),并用作比较的性能基准。

STUMP.2:stumpy.stump总共使用 2 个 CPU 执行 - 2x Intel(R) Xeon(R) CPU E5-2650 v4 @ 2.20GHz 处理器在没有 Dask 的单个服务器上与 Numba 并行。

STUMP.16:stumpy.stump总共使用 16 个 CPU 执行 - 16 个 Intel(R) Xeon(R) CPU E5-2650 v4 @ 2.20GHz 处理器在没有 Dask 的单个服务器上与 Numba 并行。

STUMPED.128:stumpy.stumped总共使用 128 个 CPU 执行 - 8x Intel(R) Xeon(R) CPU E5-2650 v4 @ 2.20GHz 处理器 x 16 个服务器,使用 Numba 并行化,并使用 Dask Distributed 进行分发。

STUMPED.256:stumpy.stumped总共使用 256 个 CPU 执行 - 8 个 Intel(R) Xeon(R) CPU E5-2650 v4 @ 2.20GHz 处理器 x 32 个服务器,使用 Numba 并行化,并使用 Dask Distributed 进行分发。

GPU-STUMP.1:stumpy.gpu_stump使用 1x NVIDIA GeForce GTX 1080 Ti GPU 执行,每块 512 个线程,200W 功率限制,使用 Numba 编译为 CUDA,并使用 Python 多处理并行化

GPU-STUMP.2:stumpy.gpu_stump使用 2x NVIDIA GeForce GTX 1080 Ti GPU 执行,每块 512 个线程,200W 功率限制,使用 Numba 编译为 CUDA,并使用 Python 多处理并行化

GPU-STUMP.DGX1:stumpy.gpu_stump使用 8x NVIDIA Tesla V100 执行,每块 512 个线程,使用 Numba 编译为 CUDA,并使用 Python 多处理并行化

GPU-STUMP.DGX2:stumpy.gpu_stump使用 16x NVIDIA Tesla V100 执行,每块 512 个线程,使用 Numba 编译为 CUDA,并使用 Python 多处理并行化

运行测试

测试在测试目录中编写并使用PyTest进行处理,并且需要coverage.py进行代码覆盖率分析。可以通过以下方式执行测试:

./test.sh

蟒蛇版本

STUMPY 支持Python 3.7+,并且由于使用 unicode 变量名/标识符,与 Python 2.x 不兼容。由于依赖关系较小,STUMPY 可能适用于较旧版本的 Python,但这超出了我们的支持范围,我们强烈建议您升级到最新版本的 Python。

获得帮助

首先,请查看 Github 上的讨论问题,看看您的问题是否已经在那里得到解答。如果没有可用的解决方案,请随时打开新的讨论或问题,作者将尝试以合理及时的方式做出回应。

贡献

我们欢迎任何形式的贡献!文档方面的帮助,尤其是扩展教程,总是受欢迎的。要贡献,请分叉项目,进行更改,然后提交拉取请求。我们将尽最大努力与您一起解决任何问题,并将您的代码合并到主分支中。

引用

如果您在科学出版物中使用过此代码库并希望引用它,请使用Journal of Open Source Software 文章

SM 法律,(2019 年)。STUMPY:用于时间序列数据挖掘的强大且可扩展的 Python 库。开源软件杂志,4(39),1504。

@article{law2019stumpy,
  author  = <s>{Law, Sean M.}</s>,
  title   = <s>{{STUMPY: A Powerful and Scalable Python Library for Time Series Data Mining}}</s>,
  journal = <s>{{The Journal of Open Source Software}}</s>,
  volume  = <s>{4}</s>,
  number  = <s>{39}</s>,
  pages   = <s>{1504}</s>,
  year    = <s>{2019}</s>
}

参考

Yeh,Chin-Chia Michael 等人。(2016) 矩阵配置文件 I:时间序列的所有对相似性连接:包括 Motifs、Discords 和 Shapelets 的统一视图。ICDM:1317-1322。关联

朱,颜,等。(2016) 矩阵配置文件 II:利用新算法和 GPU 来打破时间序列主题和连接的一亿个障碍。ICDM:739-748。关联

Yeh,Chin-Chia Michael 等人。(2017) Matrix Profile VI:有意义的多维母题发现。ICDM:565-574。关联

朱,颜,等。(2017) 矩阵配置文件 VII:时间序列链:时间序列数据挖掘的新原语。ICDM:695-704。关联

Gharghabi、Shaghayegh 等人。(2017 年)矩阵配置文件 VIII:超人类性能水平的域不可知在线语义分割。ICDM:117-126。关联

朱,颜,等。(2017) 利用新的算法和 GPU 来打破时间序列主题和连接的十万亿成对比较障碍。凯斯:203-236。关联

朱,颜,等。(2018) 矩阵配置文件 XI:SCRIMP++:以交互速度发现时间序列主题。ICDM:837-846。关联

Yeh,Chin-Chia Michael 等人。(2018) 时间序列连接、Motifs、Discords 和 Shapelets:利用矩阵配置文件的统一视图。数据最小知识光盘:83-123。关联

Gharghabi、Shaghayegh 等人。(2018) “矩阵配置文件 XII:MPdist:一种新的时间序列距离度量,允许在更具挑战性的场景中进行数据挖掘。” ICDM:965-970。关联

齐默尔曼、扎卡里等人。(2019) Matrix Profile XIV:使用 GPU 扩展时间序列主题发现以打破一天及以后的 Quintillion 成对比较。SoC '19:74-86。关联

Akbarinia、Reza 和 Betrand Cloez。(2019) 使用不同距离函数的高效矩阵轮廓计算。arXiv:1901.05708。关联

Kamgar、Kaveh 等人。(2019) 矩阵配置文件 XV:利用时间序列共识主题来查找时间序列集中的结构。ICDM:1156-1161。关联

许可证和商标

矮墩墩
版权所有 2019 德美利证券。根据 3-Clause BSD 许可条款发布。
STUMPY 是 TD Ameritrade IP Company, Inc. 的商标。保留所有权利。

项目详情