news 2026/4/18 5:29:14

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/4/18 1:20:09

高频信号布局中的铺铜技巧(AD平台)

高频信号布局中的铺铜实战指南(Altium Designer平台)你有没有遇到过这样的情况:电路原理图设计得严丝合缝,元器件选型也经过反复推敲,结果一上电就干扰不断,高速信号波形毛刺满屏,EMC测试频频超…

作者头像 李华
网站建设 2026/4/18 2:02:27

Altium Designer混合信号电路设计一文说清

Altium Designer混合信号电路设计:从仿真到落地的完整实践你有没有遇到过这样的情况?好不容易把原理图画完,PCB也布好了线,结果一上电——ADC采样噪声大得离谱,MCU时不时死机,串口通信满屏乱码。反复改板、…

作者头像 李华
网站建设 2026/4/17 13:04:09

如何将PDF、Word文档变成可对话的知识源?试试Anything-LLM

如何将PDF、Word文档变成可对话的知识源?试试Anything-LLM 在企业知识管理的日常中,你是否曾遇到这样的场景:一份上百页的年度报告摆在面前,领导突然问:“去年海外市场增长了多少?”你只能迅速翻页、逐段查…

作者头像 李华
网站建设 2026/4/15 18:02:15

大模型杀不死产品经理,但未来我们可能要做“产品界的 OnlyFans”

这是一篇合辑文,近期跟很多朋友交流了下大模型应用的感受,也有不少观点碰撞,其中很多内容实际上一脉同源,于是整理一下。 这篇文章也不讲究什么行文逻辑了,由观点引出观点,用臆测评价臆测,主打…

作者头像 李华
网站建设 2026/4/17 13:32:11

Pspice器件库开发完整指南:自定义元件封装流程

手把手教你打造专属Pspice元件:从无模型到可仿真全链路实战你有没有遇到过这样的场景?手头是一款刚发布的碳化硅MOSFET,资料齐全、参数亮眼,但翻遍官网也没找到Pspice模型;项目进度压着要仿真验证,标准库里…

作者头像 李华
网站建设 2026/4/17 18:02:51

framebuffer支持透明叠加层的技术解析

如何让嵌入式界面“透”出高级感?深入剖析 framebuffer 的透明叠加层实现你有没有在车载中控屏上见过半透明的导航提示框,像一层薄雾浮在地图之上?或者在工业控制面板里看到过动态弹出的告警窗口,背景内容若隐若现?这些…

作者头像 李华