news 2026/5/11 10:00:03

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/5/10 13:23:01

支持多模型切换的Anything-LLM究竟有多强大?

支持多模型切换的Anything-LLM究竟有多强大? 在企业知识管理日益复杂的今天,一个现实问题摆在开发者面前:如何让AI助手既能准确回答专业问题,又不把敏感数据上传到第三方API?更进一步——能否在一个系统里,…

作者头像 李华
网站建设 2026/5/11 0:52:29

如何为客户提供定制化AI文档服务?从Anything-LLM开始

如何为客户提供定制化AI文档服务?从Anything-LLM开始 在企业知识爆炸式增长的今天,员工花三小时找一份旧合同、新同事反复询问相同的入职问题、客服无法准确引用最新产品条款——这些场景每天都在真实发生。传统搜索靠关键词匹配,面对“报销流…

作者头像 李华
网站建设 2026/5/9 19:03:54

Agent工作流设计:构建自动化业务流程

Agent工作流设计:构建自动化业务流程 在企业知识管理日益复杂的今天,一个常见的尴尬场景是:新员工入职后反复询问“年假怎么休”,HR不得不再次翻出《员工手册》第12页作答;技术团队面对客户咨询时,需要在几…

作者头像 李华
网站建设 2026/5/7 11:28:21

RS232电平转换电路设计:超详细版硬件实现指南

RS232电平转换电路设计:从原理到实战的完整硬件指南 在工业现场,你是否曾遇到这样的尴尬?——手握一块性能强大的STM32开发板,却无法与一台老式温控仪通信。串口线接上后,数据要么乱码、要么完全无响应。问题出在哪&am…

作者头像 李华
网站建设 2026/4/22 11:44:14

告别AI黑话!程序员入门大模型的必备术语详解,建议收藏

本文以通俗易懂的方式解释了AI和大模型领域的15个核心术语,包括序列化、反序列化、解析、解析器、块、词元、向量、嵌入、检索增强生成(RAG)、提示词、温度、TOP P、LLM、预训练和微调等。作者旨在帮助初学者理解这些"行业黑话",消除学习障碍&…

作者头像 李华
网站建设 2026/5/11 2:51:55

日程安排建议:智能协调多方时间空档

智能协调多方时间空档:当AI成为你的日程顾问 在一家跨国科技公司,项目经理小李每周最头疼的事不是写报告,而是约一次全员参会的项目同步会。团队成员分布在旧金山、柏林和上海,各自使用不同的日历工具,沟通靠邮件来回确…

作者头像 李华