Question

我正在尝试在宏观经济时间序列的趋势和周期（1985）中复制结果。如果仅使用局部线性模型，则能够获得相同的结果，但是在添加固定的周期性分量时却无法获得相同的结果。尽管我不确定，但我怀疑这与对所有状态使用完全漫射先验有关，而不是仅对局部线性分量起作用。为了保持一定程度的灵活性，我想使用statsmodels关键字向known输入缩写，并以update方法进行更新。现在，我不必担心使用近似扩散。实现此目标的最佳方法是什么？

Answer 1

从v0.10开始，通过创建自己的Initialization实例，您可以在设置初始化方面获得很大的灵活性。使用这种方法，可以将每个状态或状态块独立设置为使用以下可用方法之一：known，stationary，approximate_diffuse或diffuse。

这里是局部级别模型的示例，但其中不规则分量被建模为AR（1）。第一个状态是AR（1），我指定了stationary初始化。第二个状态是级别，我使用known初始化通过最大似然估计初始级别。第三种状态是斜率，我将其初始化设置为diffuse。

import numpy as np
import pandas as pd
import statsmodels.api as sm

from statsmodels.tsa.statespace.initialization import Initialization
from statsmodels.tsa.statespace.tools import (
    constrain_stationary_univariate,
    unconstrain_stationary_univariate)

class LocalLevelAR1(sm.tsa.statespace.MLEModel):
    def __init__(self, endog):
        super().__init__(endog, k_states=3, k_posdef=3)

        self['design', 0, :2] = 1
        self['transition', 1:, 1:] = np.array([[1., 1.],
                                               [0., 1.]])
        self['selection'] = np.eye(3)

        init = Initialization(3)
        init.set(0, 'stationary')
        init.set(1, 'known', stationary_cov=[[0]])
        init.set(2, 'diffuse')
        self.ssm.initialization = init

    @property
    def state_names(self):
        return ['ar1', 'level', 'slope']

    @property
    def param_names(self):
        return ['level0', 'phi', 'sigma2.ar1', 'sigma2.level', 'sigma2.slope']

    @property
    def start_params(self):
        return np.r_[0, 0, 1e-5, 1e-5, 1e-5]

    def transform_params(self, unconstrained):
        constrained = unconstrained.copy()
        constrained[1] = constrain_stationary_univariate(unconstrained[1:2])
        constrained[2:5] = constrained[2:5]**2
        return constrained

    def untransform_params(self, constrained):
        unconstrained = constrained.copy()
        unconstrained[1] = unconstrain_stationary_univariate(constrained[1:2])
        unconstrained[2:5] = constrained[2:5]**0.5
        return constrained

    def update(self, params, **kwargs):
        params = super().update(params, **kwargs)

        self['transition', 0, 0] = params[1]
        self['state_cov'] = np.diag(params[2:5])
        self.ssm.initialization.set(1, 'known', constant=params[:1], stationary_cov=[[0.]])

dta = sm.datasets.macrodata.load_pandas().data
dta.index = pd.period_range(start='1959Q1', end='2009Q3', freq='Q')

mod = LocalLevelAR1(dta['infl'])
res = mod.fit()

print(res.summary())

用户在状态空间模型中提供的初始状态

1 个答案: