aeon：开源时间序列机器学习工具包

aeon：开源时间序列机器学习的新星

在当今数据驱动的世界中，时间序列分析和预测扮演着越来越重要的角色。无论是在金融、医疗、工业还是其他领域，能够准确分析和预测时间序列数据都是至关重要的。为了满足这一需求，aeon应运而生，成为了时间序列机器学习领域的一颗新星。

aeon logo

aeon简介

aeon是一个开源的时间序列机器学习工具包，旨在为研究人员和数据科学家提供一个全面、高效且易于使用的平台。它不仅兼容广受欢迎的scikit-learn库，还提供了最新的时间序列算法和经典技术，支持预测、分类、聚类等多种学习任务。

aeon的目标是成为时间序列算法的综合库，包括最新的研究成果，同时提供高效的实现和与其他时间序列包的接口，为算法比较提供统一的框架。

aeon的主要特点

兼容性：与scikit-learn完全兼容，可以无缝集成到现有的机器学习工作流程中。
全面性：提供广泛的时间序列算法，包括最新的研究成果和经典方法。
高效性：使用numba等技术进行优化，确保算法的高效实现。
灵活性：支持多种时间序列学习任务，如预测、分类、聚类等。
可扩展性：提供清晰的API，方便用户扩展和定制算法。
文档完善：提供详细的文档和示例，帮助用户快速上手。

安装aeon

aeon要求Python版本在3.9或以上。安装aeon最简单的方式是通过pip：

pip install aeon

如果你想安装所有可选依赖，可以使用：

pip install aeon[all_extras]

对于开发者，可以从GitHub源码安装，详细说明可以在开发者安装指南中找到。

使用aeon

让我们通过一些简单的例子来了解如何使用aeon进行时间序列分类和聚类。

时间序列分类

import numpy as np
from aeon.classification.distance_based import KNeighborsTimeSeriesClassifier

# 准备数据
X = [[[1, 2, 3, 4, 5, 5]],  # 3D数组示例（单变量）
     [[1, 2, 3, 4, 4, 2]],  # 三个样本，一个通道，六个时间点
     [[8, 7, 6, 5, 4, 4]]]
y = ['low', 'low', 'high']  # 每个样本的类别标签
X = np.array(X)
y = np.array(y)

# 创建并训练分类器
clf = KNeighborsTimeSeriesClassifier(distance="dtw")
clf.fit(X, y)

# 预测新数据
X_test = np.array(
    [[[2, 2, 2, 2, 2, 2]], [[5, 5, 5, 5, 5, 5]], [[6, 6, 6, 6, 6, 6]]]
)
y_pred = clf.predict(X_test)
print(y_pred)  # 输出：['low' 'high' 'high']

在这个例子中，我们使用了基于动态时间规整（DTW）距离的K近邻时间序列分类器。这个分类器可以很容易地替换为回归器，标签也可以替换为数值目标，展示了aeon在不同任务和数据集之间的灵活性。

时间序列聚类

import numpy as np
from aeon.clustering import TimeSeriesKMeans

# 准备数据
X = np.array([[[1, 2, 3, 4, 5, 5]],  # 3D数组示例（单变量）
     [[1, 2, 3, 4, 4, 2]],  # 三个样本，一个通道，六个时间点
     [[8, 7, 6, 5, 4, 4]]])

# 创建并训练聚类器
clu = TimeSeriesKMeans(distance="dtw", n_clusters=2)
clu.fit(X)

# 获取训练数据的聚类标签
print(clu.labels_)  # 输出：array([0, 0, 1])

# 预测新数据的聚类
X_test = np.array(
    [[[2, 2, 2, 2, 2, 2]], [[5, 5, 5, 5, 5, 5]], [[6, 6, 6, 6, 6, 6]]]
)
print(clu.predict(X_test))  # 输出：array([1, 0, 0])