如何使用pyramid.arima.auto_arima()
时间: 2024-10-23 16:21:26 浏览: 29
`pyramid.arima.auto_arima()`是`prophet-arima`库(由Facebook开发)的一部分,它提供了一个方便的接口来自动估计ARIMA模型的超参数,包括p(自回归阶数)、d(差分阶数)和q(移动平均阶数)。这个函数是基于信息准则(如AIC或BIC)来优化模型复杂度的。
以下是使用`auto_arima()`的基本步骤:
1. **安装库**:
如果还没有安装,首先需要使用pip安装`prophet-arima`:
```
pip install pyramid-arima
```
2. **导入库**:
```python
from pyramid.arima import auto_arima
```
3. **加载数据**:
确保你的数据是一个包含日期索引和数值观测值的Series或DataFrame。
```python
import pandas as pd
data = pd.read_csv('your_data.csv', index_col='date_column')
```
4. **调用函数**:
```python
model = auto_arima(data['value_column'], start_p=1, start_q=1,
max_p=3, max_q=3, m=12, d=None, start_d=0, seasonal=True,
stepwise=True, error_action='ignore', suppress_warnings=True)
```
参数说明:
- `start_p` 和 `start_q`: 初始的p和q值。
- `max_p` 和 `max_q`: 最大的p和q值搜索范围。
- `m` (季节数): 如果数据存在明显的季节性,指定季节周期。
- `d` 或 `start_d` 和 `seasonal`: 如果数据需要差分处理,可以选择开始差分的阶数或默认为0。
- `stepwise`: 是否允许步进式搜索超参数。
- `error_action` 和 `suppress_warnings`: 控制错误处理和警告显示。
5. **获取结果**:
调用`fit()`方法训练模型,然后获取最佳参数和拟合结果:
```python
model_fit = model.fit()
```
6. **模型评估**:
可以使用`model_fit.aic`或`model_fit.bic`等属性来评估模型性能。
7. **预测**:
若要进行未来预测,可以用`forecast()`方法:
```python
forecast = model_fit.forecast(steps=10)
```
阅读全文