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
相关产品推荐
相关产品推荐

