You need to enable JavaScript to run this app.
优惠活动
大模型
产品
解决方案
定价
更多

StatsForecast AutoARIMA拟合速度骤降,寻求技术解决方案

StatsForecast运行速度骤降100倍的问题排查求助

使用Nixtla的StatsForecast时遇到严重性能问题,此前运行速度正常,今日突然变慢约100倍,严重影响工作进度。

官方文档提到,在100万条序列上拟合10个模型可在5分钟内完成,但运行以下代码时,电脑耗时超30分钟仍未结束:

import random 
import pandas as pd
from statsforecast import StatsForecast
from statsforecast.models import AutoARIMA
import time

random.seed(1)
points = 500
test_data = pd.DataFrame({
    "ds": pd.date_range(start="2024-06-01", periods=points, freq="W"),
    "unique_id": "test",
    "y": random.choices(range(0, 100), k=points)
})
model = StatsForecast(
    models=[AutoARIMA(season_length=52)],
    freq="W",
    n_jobs=-1,
)
start = time.time()
results = model.forecast(df=test_data, h=6)
print(time.time() - start)
results= model.forecast(df=test_data, h=6)
results

已进行以下测试,但均未解决问题:

  • 重启电脑
  • 从零创建Python 3.10环境,脱离PyCharm直接通过终端运行
  • 调整核心数,但无明显效果,CPU使用率从未超过100%
  • 在其他机器测试:使用相同版本statsforecast 1.7.5,运行速度明显更快

当前使用的PC配置:Ubuntu 20.04系统,16GiB内存,Intel Core i7 1.8Ghz x8处理器。是否是配置问题导致?恳请各位提供解决思路!

编辑说明:已修复随机种子并添加计时功能。

内容的提问来源于stack exchange,提问作者aarcas

相关产品推荐
方舟 Agent Plan

超全模态模型 × Harness 升级,最新支持 Deepseek-V4.1-Flash、GLM-5.3 系列、Doubao-Seedream-5.0-pro、Kimi-K3 (部分), 限时 9.9 元起

最近更新时间:2026.06.21 18:57:04