news 2026/4/11 18:49:39

Python 第三方库:darts(现代化时间序列建模与预测框架)

作者头像

张小明

前端开发工程师

1.2k 24
文章封面图
Python 第三方库:darts(现代化时间序列建模与预测框架)

darts 是一个专门用于时间序列分析、建模与预测的 Python 库,提供统一而高层的 API,集成了统计模型(如 ARIMA)、机器学习模型(如 LightGBM)、深度学习模型(如 RNN、Transformer、N-BEATS、TCN 等)以及时间序列特有的处理工具(如窗口生成、协变量管理、回测、误差评估等)。

darts 的设计目标是让时间序列预测像使用 scikit-learn 一样简单,同时让深度学习模型的构建更加易用。

安装:

pip install darts

如需使用深度学习模型:

pip install darts[u]

常见应用场景:

(1)时间序列预测(Forecasting)

使用 ARIMA、RNN、Transformer、N-BEATS 等模型进行未来值预测。

(2)多元时间序列建模

支持多变量输入、协变量(covariates)、滞后特征等复杂数据结构。

(3)回测与模型评估

内置回测、滚动预测、交叉验证与误差指标。

(4)异常检测

支持时间序列的异常值检测及预测残差分析。

(5)时间序列处理

提供插值、缩放、缺失值填补、时间对齐等操作。

(6)深度学习时间序列模型

内置 PyTorch 实现的专业算法,如:

• RNNModel

• TCNModel

• NBEATSModel

• TransformerModel

• TFTModel(Temporal Fusion Transformer)

拥有高性能与易用性的平衡。

◆ ◆

核心概念

1、TimeSeries 对象(核心数据结构)

darts 所有模型的输入与输出均为 TimeSeries 类型。

它包含:

• 时间索引(DatetimeIndex 或 RangeIndex)

• 数据列(可以是一列或多列)

• 可选的附加维度(例如多个样本)

创建方式示例:

from darts import TimeSeriesimport pandas as pd df = pd.read_csv("data.csv")series = TimeSeries.from_dataframe(df, "date", "value")

2、预测模型(Models)

darts 的模型分为:

• 统计模型(ARIMA、Exponential Smoothing、Theta 等)

• 机器学习模型(RegressionModel)

• 深度学习模型(NBEATS、RNN、TFT 等)

• EnsembleModel(用于模型集成)

每一个模型都提供统一的 API:

model = SomeModel(...)model.fit(series)forecast = model.predict(n)

3、协变量(Covariates)

时间序列预测常需要外生变量,如:

• 天气

• 节假日

• 商品价格

• 滞后窗口

darts 支持三类协变量:

• past_covariates:过去已知的数据

• future_covariates:未来已知的数据(如节日表)

• static_covariates:不会随时间变化的特征

4、回测(Backtesting)

darts 内置回测工具,通过滑动窗口自动评估模型表现:

from darts.metrics import mape model.backtest( series, forecast_horizon=12, metric=mape)

5、误差指标(Metrics)

支持:

• MAE

• RMSE

• MAPE

• SMAPE

• DTW 距离

• R2 score

用于评估预测效果。

◆ ◆

应用举例

例 1:加载数据并创建 TimeSeries

from darts import TimeSeriesimport pandas as pd df = pd.read_csv("sales.csv")series = TimeSeries.from_dataframe(df, "date", "sales")

例 2:使用 ARIMA 进行预测

from darts.models import ARIMA model = ARIMA()model.fit(series)forecast = model.predict(12)forecast.plot(label="forecast")

例 3:使用深度学习模型(N-BEATS)

from darts.models import NBEATSModel model = NBEATSModel( input_chunk_length=24, output_chunk_length=12, n_epochs=50,) model.fit(series)prediction = model.predict(12)

例 4:使用协变量进行多元预测(RNN)

from darts.models import RNNModel model = RNNModel( model="LSTM", input_chunk_length=30, output_chunk_length=7) model.fit(series, future_covariates=weather_series)forecast = model.predict(7, future_covariates=weather_series)

例 5:回测评估模型

from darts.metrics import smape error = model.backtest( series, forecast_horizon=12, metric=smape,)print("SMAPE:", error)

◆ ◆

常用函数与类速览

TimeSeries.from_dataframe(df, time_col, value_cols)

从 DataFrame 创建时间序列对象。

参数:

df:pandas DataFrame。

time_col:时间列名。

value_cols:数值列名或列表。

返回:TimeSeries 实例。

model.fit(series, **covariates)

训练模型。

参数:

series:目标时间序列。

past_covariates:可选,历史协变量。

future_covariates:可选,未来协变量。

返回:无。

model.predict(n, **covariates)

预测未来 n 步。

参数:

n:预测步数。

future_covariates:未来外生变量。

返回:TimeSeries 类型的预测结果。

model.backtest(series, forecast_horizon, metric)

对时间序列执行滑动窗口回测。

参数:

series:目标时间序列。

forecast_horizon:一次预测步数。

metric:误差函数。

返回:float(误差值)。

TimeSeries.plot()

快速绘制时间序列。

参数: 无。

返回: 无(生成图形)。

📘 小结

darts 是一套现代化的时间序列分析框架,具有以下优势:

• 统一的 API,几十种模型可无缝切换

• 内置深度学习模型,支持 PyTorch 训练

• 支持协变量、窗口生成、多元时间序列

• 强大的回测工具与评估指标

• TimeSeries 统一数据结构,简化时间序列处理

• 可从快速原型扩展到企业级模型

无论你是做金融预测、供应链预测,还是做科学研究或智能调度,darts 都提供稳定、灵活而完整的工具链。

“点赞有美意,赞赏是鼓励”

版权声明: 本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若内容造成侵权/违法违规/事实不符,请联系邮箱:809451989@qq.com进行投诉反馈,一经查实,立即删除!
网站建设 2026/3/29 7:50:41

私有化部署首选:Anything-LLM企业级RAG系统详解

私有化部署首选:Anything-LLM企业级RAG系统详解 在金融、医疗和法律等行业,数据安全早已不是“加分项”,而是生存底线。当大语言模型席卷各行各业时,这些领域的企业却不得不面对一个尴尬局面:公有云AI服务虽然强大&…

作者头像 李华
网站建设 2026/4/3 4:15:44

工业环境EMC兼容PCB设计案例通俗解释

工业级EMC设计实战:从原理到落地的PCB避坑指南你有没有遇到过这样的场景?一块功能完全正常的控制板,放到现场却频繁重启;ADC采集的数据跳得像心电图,可实验室里明明很稳定;RS485通信距离一长就丢包&#xf…

作者头像 李华
网站建设 2026/4/3 7:04:46

一键部署Anything-LLM,快速接入GPU算力与Token服务

一键部署Anything-LLM,快速接入GPU算力与Token服务 在企业知识管理日益复杂的今天,如何让员工快速获取分散在PDF、文档和内部系统中的信息,成为提升效率的关键瓶颈。更棘手的是,许多公司既希望享受大语言模型(LLM&…

作者头像 李华
网站建设 2026/4/10 16:54:53

Agent驱动的工作流开发新范式:颠覆传统编程,效率提升10倍

本文介绍了一种基于Agent驱动的工作流开发新模式,颠覆传统"先建模,后执行"的思维,转向"先定义目标,后自组织执行"。该模式具有参数自组织、上下文自感知、无需固定流程编排等优势,能大幅简化开发流…

作者头像 李华
网站建设 2026/4/11 15:35:15

EasyEDA平台下嘉立创PCB布线核心要点解析

从设计到制造:在 EasyEDA 上做好嘉立创 PCB 布线的实战指南 你有没有遇到过这种情况?花了一周时间画好原理图、布完板子,兴冲冲导出 Gerber 文件上传嘉立创,结果 DFM 检测报告弹出十几个红色警告——“间距不足”、“焊盘太小”、…

作者头像 李华
网站建设 2026/4/11 6:18:29

基于LVGL的触摸屏厨房电器UI设计案例

从零打造厨房电器智能触控界面:LVGL实战全解析 你有没有过这样的经历?站在微波炉前,盯着那排密密麻麻的机械按钮,反复按“30秒”五次只为加热两分钟;或者在电饭煲上翻三页菜单才找到“快煮”模式。这些繁琐操作的背后…

作者头像 李华