Loading [MathJax]/jax/output/CommonHTML/fonts/TeX/fontdata.js

logo

Prophet

wangzf / 2022-04-30

目录

论文简介

论文：Forecast at Scale

代码：https://github.com/facebook/prophet

文档：https://facebook.github.io/prophet/

历史研究和瓶颈

ARMA 模型在时间序列的趋势和季节上没有很好的处理， ETA(指数平滑，Exponential Smoothing) 和 SNAIVE(季节朴素法，Seasonal Navie)虽然能抓住季节性但不能抓到更长周期性。

论文贡献

Facebook 开源了一个时间序列预测的算法，叫做 fbprophet。 从官网的介绍来看，Facebook 所提供的 Prophet 算法不仅可以处理时间序列存在一些异常值的情况， 也可以处理部分缺失值的情形，还能够几乎全自动地预测时间序列未来的走势。

从论文上的描述来看，这个 Prophet 算法是基于时间序列分解和机器学习的拟合来做的， 其中在拟合模型的时候使用了 pyStan 这个开源工具，因此能够在较快的时间内得到需要预测的结果。

从整体的介绍来看，如果是一般的商业分析或者数据分析的需求， 都可以尝试使用这个开源算法来预测未来时间序列的走势。 对于商业分析等领域的时间序列，Prophet 可以进行很好的拟合和预测， 但是对于一些周期性或者趋势性不是很强的时间序列， 用 Prophet 可能就不合适了(例如：移动平均)。

但是，Prophet 提供了一种时序预测的方法， 在用户不是很懂时间序列的前提下都可以使用这个工具得到一个能接受的结果。 具体是否用 Prophet 则需要根据具体的时间序列来确定。

模型定义

2018 年 Facebook 提出的 Prophet，主要是让分析师将业务知识融入到时序建模当中， 同时能批量建模。

概率视角下的贝叶斯线性回归

从概率的角度来讲，时间序列是随机过程在时间方向上的一次采样， 所以时间序列模型的建模过程是期望通过这样的采样数据来学习模型。 同理，机器学习模式的有监督学习的训练样本是 P(y|x) 在空间方向上的采样， 只不过相对于时间不能重复来说，机器学习的采样在空间上可以不断重复。

先从一个简单的线性回归开始，线形回归有概率学派的视角和贝叶斯学派的视角。 从频率学派的视角，做的是一个点估计，而贝叶斯学派是一个概率估计， 也就是说目标值 y 不再是一个被估计的单个值，而是假设从某个分布(比如正态分布)中提取出来的。

一元线性回归的假设空间可表示如下：

F={P(y|x,θ)=1√2πσexp(−(y−x⋅θ)22σ2),θ∈Rm+1}

简单解释：x 是样本向量，当输入了样本 x 进去之后， 就能得到每个 y 的概率分布是多少。如果是频率学派，学习的是这条线的一个表达式， 但是目前贝叶斯学派，学到的是每个 x 在 y 轴上的分布。 在分析上面表达式的时候，不难想到，如果通过高斯分布去分析的话， 对于 y 的分布，它们的形状，也就是方差 σ 是一样的， 但是期望值 μ 是随着 x,θ 进行变化，而且是沿着一条线进行变化。

在概率视角下，和频率学派一样，要去学习参数 θ。

频率学派学习方法

频率学派的学习方法是通过训练数据和最大似然估计去得到 θ 的具体值。 p(x|θ) 中的 θ 是一个常量，对于 N 个观测来说观测集的概率为：

p(x|θ)=iidN∏i=1p(xi|θ)

为了求 θ 的大小，采用最大对数似然(MLE)方法：

θMLE=argmaxθ log p(x|θ)=iidargmaxθN∑i=1log p(xi|θ)

贝叶斯学派学习方法

贝叶斯学派中 θ 是一个随机变量而不是一个固定的值，它由一个先验分布进行约束，下面是贝叶斯线形回归：

假设 (θ1,θ2,⋯,θm) 是服从均值为 0、 协方差矩阵为对角矩阵 diag(λ) 的多元高斯分布 N(0,diag(λ))， 这是一个先验分布。通过观察到的采样数据（训练样本）不断对先验分布进行更新，从而得到后验分布。

于是，要求解的是在观察到训练样本 T=(xi,yi),i=1,⋯,n 之后， (θ1,θ2,⋯,θm) 所服从的后验分布 Ppost(θ|T)， 这是一个条件概率。

Ppost(θ|T)=n∏i=1P(yi|xi,θ)P(θ1,θ2,⋯,θm)=n∏i=1P(yi|xi,θ)m∏j=11√2πλexp(−θ2j2λ)

在已知最大后验分布的情况下，求解参数 θ 可以使用最大后验估计或者贝叶斯估计两种方法：

在求解参数 θ 的过程中，还有一种方式是做贝叶斯估计，但在一般情况下， 很难直接求贝叶斯估计的后验分布的解析解，所以一般会使用后验分布的采样值，采样的方法是使用 MCMC 采样。 然后得到的是 θ 的采样值，这样预测值y就也是一个不固定的预测值， 得到的也是 y 的分布。这一点开始我们可以往 Prophet 走了，记得 Prophet 里面有一个 y_upper, y_lower 吗， 这不就是说明 y 也是有一个范围吗，而不是完完全全的一个确定值。

时间序列分解

在时间序列分析中，常见的分析方法叫做时间序列的分解(Decomposition of Time Series)， 时间序列分解的原理是将时间序列 y(t) 分为三个部分:

根据季节性项 s(t) 的变化，分解方法可分为两种:

Prophet 是将时间序列看成是一个关于 t 的函数，用拟合函数曲线的方式进行预测， Prophet 的输入只需要 ts 和 y。所以核心公式是：

y(t)=trend(t)+seasonality(t)+holiday(t)+error(t)

一般来说，在实际生活和生产环节中，除了季节项、趋势项、剩余项之外，通常还有节假日的效应。 所以，在 Prophet 算法里面，作者同时考虑了以上四项，Prophet 算法就是通过拟合这几项， 然后最后把它们累加起来就得到了时间序列的预测值。 也就是:

y(t)=g(t)+s(t)+h(t)+ϵt

其中:

趋势项模型

趋势项模型：g(t)

在 Prophet 算法里面，趋势项分为两种：

g(t)=C1+e−k(t−m)

g(t)=kt+m

饱和趋势项

首先，介绍一下基于逻辑回归函数(logistic function)的饱和趋势项(Nonlinear，Saturating Growth)。

回顾逻辑回归函数，一般都会想起 Sigmoid 函数形式：

σ(x)=11+e−x

它的导函数是 σ′(x)=σ(x)⋅(1−σ(x))， 并且 limx→+∞σ(x)=1， limx→−∞σ(x)=0。

从 Sigmoid 的函数表达式可以看出，它满足以下的微分方程：y′=y(1−y)。 那么，如果使用分离变量法来求解微分方程 y′=y(1−y) 就可以得到： y′y+y′1−y=1 lny1−y=1 y=11+ke−x

如果增加一些参数的话，逻辑回归就可以改写成：

g(t)=C1+e−k(t−m)

其中：

当 C=1，k=1，m=0 时，恰好就是 Sigmoid 函数的形式。 在现实情况下，函数 g(x) 的三个参数 C,k,m 不可能都是常数， 而很有可能是随着时间的迁移而变化的，因此，在 Prophet 里面，作者考虑把这三个参数全部换成了随着时间变化的函数， 即：C=C(t)，k=k(t)，m=m(t)。

除此之外，在现实的时间序列中，曲线的走势肯定不会一直保持不变， 在某些特定的时候或者有着某种潜在的周期曲线会发生变化， 这种时候，就有学者会去研究变点检测(change point detection)。 例如下面的这幅图的 t∗1，t∗2 就是时间序列的两个变点

img

在 Prophet 里面，是需要设置变点的位置的，而每一段的趋势和走势也是会根据变点的情况而改变的。 在程序里面有两种方法，一种是通过人工指定的方式指定变点的位置；另外一种是通过算法来自动选择。

在默认的函数里面，Prophet 会选择 n_changepoints = 25 个变点，然后设置变点的范围是前 changepoint_range = 80%， 也就是在时间序列的前 80% 的区间内会设置变点。通过 forecaster.py 里面的 set_changepoints 函数可以知道， 首先要看一些边界条件是否合理，例如时间序列的点数是否少于 n_changepoints 等内容； 其次如果边界条件符合，那变点的位置就是均匀分布的，这一点可以通过 np.linspace 这个函数看出来。

下面假设已经设置了 S 个变点了，并且变点的位置是在时间戳 sj,1≤j≤S 上， 那么在这些时间戳上，就需要给出增长率的变化，也就是在时间戳 sj 上发生的增长量变化率(change in rate)。 假设有这样一个向量：δ=(δ1,…,δS)∈RS， 其中 δj 表示在时间戳 sj 上的增长率的变化量，如果初始增长率使用 k 表示， 那么在时间戳 t 上的增长率就是 k+∑j:t>sjδj， 通过一个指示函数 aj(t)∈{0,1}S：

aj(t)={1,if t≥sj,0,otherwise

表示在时间戳 t 上面的增长率就是：

k(t)=k+a(t)Tδ

一旦变化率 k(t) 确定了，另一个参数 m(t) 也要随之确定。 在这里需要把线段的边界处理好，因此通过数学计算可以得到：

γj=(sj−m−∑l<jγl)⋅(1−k+∑l<jδlk+∑l≤jδl)

m(t)=m+a(t)Tγ

所以，分段的逻辑回归增长模型就是：

g(t)=C(t)1+e−k(t)(t−m(t))=C(t)1+e−(k+a(t)Tδ)⋅(t−(m+a(t)Tγ))

其中：

在逻辑回归函数里面，有一个参数是需要提前设置的，那就是 Capacity(cap)，也就是所谓的 C(t)， 在使用 Prophet 的 growth = 'logistic' 的时候，需要提前设置好 C(t) 的取值才行。

分段线性函数

Piecewise Linear Function

下面介绍一下基于分段线性函数的趋势项。 众所周知，线性函数指的是 y=kx+b，而分段线性函数指的是在每一个子区间上， 函数都是线性函数，但是在整段区间上，函数并不完全是线性的。

因此，基于分段线性函数的模型形如：

g(t)=(k+a(t)δ)⋅t+(m+a(t)Tγ)

其中：

逻辑回归函数与分段线性函数最大的区别就是 γ 的设置不一样，在分段线性函数中，

γ=(γ1,…,γS),γj=−sjδj

在 Prophet 的源代码中，forecast.py 这个模块里面包含了最关键的步骤。

而在 piecewise_linear 这个函数中，是不需要 capacity 这个指标的， m = Prophet() 这个函数默认的使用 growth = 'linear' 这个增长函数， 也可以写作 m = Prophet(growth = 'linear')。

变点的选择

Changepoint Selection

在介绍变点之前，先要介绍一下 Laplace 分布

Laplace 分布的概率密度函数为：

f(x|μ,b)=e−|x−μ|b2b

其中：

在 Prophet 算法中，是需要给出变点的位置、个数，以及增长的变化率的。 因此，有三个比较重要的指标：

在 Prophet 的默认的场景下，变点的选择是基于时间序列的前 80% 的历史数据， 然后通过等分的方法找到 25 个变点（change points）， 而变点的增长率是满足 Laplace 分布 δj∼Laplace(0,0.05) 的。 因此，当 τ 趋近于零的时候，δj 也是趋向于零的， 此时的增长函数将变成全段的逻辑回归函数或者线性函数。这一点从 g(t) 的定义可以轻易地看出。

对未来的预估

Trend Forecast Uncertainty

从历史上长度为 T 的数据中，可以选择出 S 个变点， 它们所对应的增长率的变化量是 σj∼Laplace(0,τ)。 此时需要预测未来，因此也需要设置相应的变点位置。

从代码中看，在 forecaster.py 的 sample_predictive_trend() 函数中， 通过 Poisson 分布等概率分布方法找到新增的 changepoint_ts_new 的位置， 即未来时间戳的变点位置是通过 Poission 过程采样得到的，结果依然是未来变点出现的概率和历史相似。 然后与 changepoint_t 拼接在一起就得到了整段序列的 changepoint_ts。

下面第一行代码的 1 保证了 changepoint_ts_new 里面的元素都大于 changepoint_ts 里面的元素。 除了变点的位置之外，也需要考虑 δ 的情况。 这里令 λ=∑Sj=1|δj|S， 于是新的增长率的变化量就是按照下面的规则来选择的：对任意的 j>T：

{δj=0,with probability T−STδj∼Laplace(0,λ),with probability ST

changepoint_ts_new = 1 + np.random.rand(n_changes) * (T - 1)
changepoint_ts = np.concatenate((
    self.changepoints_t, 
    changepoint_ts_new
))

详细解释见：https://zhuanlan.zhihu.com/p/181708348

假设未来出现变点的频率和历史一样。 历史一共 T 个点，其中有 S 个是变点，因此未来也应该是 S/T 的频率出现变点。

季节性趋势

季节性趋势：s(t)

几乎所有的时间序列预测模型都会考虑这个因素， 因为时间序列通常会随着天、周、月、年等季节性的变化而呈现季节性的变化， 也称周期性的变化。

对于周期函数而言，能够马上联想到的就是正弦、余弦函数。 而在数学分析中，区间内的周期性函数是可以通过正弦和余弦的函数来表示的： 假设 f(x) 是以 2π 为周期的函数，那么它的傅立叶级数就是：

a0+∞∑n=1(ancos(nx)+bnsin(nx))

在论文中，作者使用傅立叶级数来模拟时间序列的周期性。假设 P 表示时间序列的周期， P=365.25 表示以年为周期，P=7 表示以周为周期。它的傅立叶级数的形式都是：

s(t)=N∑n=1(ancos(2πntP)+bnsin(2πntP))

就作者的经验而言，对于以年为周期的序列(P=365.25)而言，N=10； 对于以周为周期的序列(P=7)而言，N=3。这里的参数可以形成列向量：

β=(a1,⋯aN,b1,⋯bN)T

当 N=10 时：

Xt=[cos(2π×1×t365.25),…,cos(2π×10×t365.25),sin(2π×1×t365.25),…,sin(2π×10×t365.25)]

当 N=3 时：

Xt=[cos(2π×1×t7),…,cos(2π×3×t7),sin(2π×1×t7),…,sin(2π×3×t7)]

因此，时间序列的季节项就是：

s(t)=X(t)β

其中：

节假日效应

节假日效应(Holidays and Events effect)：h(t)

在现实场景中，除了周末，同样有很多节假日，而且不同的国家有着不同的假期。 在 Prophet 里面，通过维基百科里面对各个国家的节假日的描述， hdays.py 收集了各个国家的特殊节假日。除了节假日之外， 用户还可以根据自身的情况来设置必要的假期，例如 The Super Bowl，双十一等。

由于每个节假日对时间序列的影响程度不一样，例如春节，国庆节则是七天的假期， 对于劳动节等假期来说则假日较短。因此，不同的节假日可以看成相互独立的模型， 并且可以为不同的节假日设置不同的前后窗口值，表示该节假日会影响前后一段时间的时间序列。

对与第 i 个节假日来说，Di 表示该节假日的前后一段时间。 为了表示节假日效应，需要一个相应的指示函数(indicator function)， 同时需要一个参数 κi 来表示节假日的影响范围。 假设有 L 个节假日，那么：

h(t)=Z(t)κ=L∑i=1κi⋅1{t∈Di}

其中：

时序模型拟合

时间序列已经可以通过趋势项、季节项、节假日项来构建了:

y(t)=g(t)+s(t)+h(t)+ϵt

下一步只需要拟合函数就可以了，在 Prophet 里面， 作者使用了 pyStan 这个开源工具中的 L-BFGS 方法来进行函数的拟合。 具体可以参考 forecast.py 里面的 stan_init 函数。

Prophet 使用

Prophet 安装

# install on PyPI
$ pip install prophet

# install using conda-forge
$ conda install -c conda-forge prophet

Prophet 简单使用

Prophet 时间序列格式

Prophet 所需要的时间序列也是这种格式的，根据官网的描述，只要用 csv 文件存储两列即可:

ds y
0 2017-10-20 22:00:00 9.5
1 2017-10-20 22:01:00 8.5
2 2017-10-20 22:02:00 8.1
3 2017-10-20 22:02:00 8.0
4 2017-10-20 22:02:00 7.8

通过 Prophet 的计算，可以计算出如下的数值:

ds yhat yhat_lower yhat_upper
3265 2017-10-20 22:00:00 8.19 7.48 8.96
3266 2017-10-20 22:01:00 8.52 7.79 9.26
3267 2017-10-20 22:02:00 8.31 7.55 9.04
3268 2017-10-20 22:02:00 8.14 7.42 8.86
3269 2017-10-20 22:02:00 8.15 7.39 8.88

使用示例

导入工具库:

import pandas as pd
from fbprophet import Prophet

数据读入:

df = pd.read_csv("./data/example_wp_log_peyton_manning.csv")
df.head()

建立模型:

m = Prophet()
m.fit(df)

模型预测:

future = m.make_future_dataframe(periods = 365)
future.head()
future.tail()

forecast = m.predict(future)
forecast[["ds", "yhat", "yhat_lower", "yhat_upper"]].tail()

forecast 可视化:

fig1 = m.plot(forecast)

forecast 组件可视化:

fig2 = m.plot_components(forecast)

from fbprophet.plot import plot_plotly, plot_components_plotly
import plotly.offline as py

py.init_notebook_mode()
fig3 = plot_plotly(m, forecast)
fig4 = plot_components_plotly(m, forecast)
py.iplot(fig)

数据操作

因为 Prophet 所需要的两列名称是 ds 和 y，其中，ds 表示时间戳， y 表示时间序列的值，因此通常来说都需要修改 pandas.DataFrame 的列名字

df = df.rename(columns = {
    "timestamp": "ds",
    "value": "y",
})

df["ds"] = pd.to_datetime(df["ds"], unit = "s")

df["y"] = (df["y"] - df["y"].mean()) / (df["y"].std())

模型拟合

先初始化模型，然后拟合模型

# 初始化模型
model = Prophet()

# 拟合模型
model.fit(X_train)

模型预测

# 计算预测值: periods 表示需要预测的点数，freq 表示时间序列的频率. 
future = model.make_future_dataframe(periods = 30, freq = "min")
forecast = model.predict(future)

# 画出预测图
model.plot(forecast)
# 画出时间序列的分量
model.plot_components(forecast)
# 绘制更详细的指标
x1 = forecast['ds']
y1 = forecast['yhat']
y2 = forecast['yhat_lower']
y3 = forecast['yhat_upper']
plt.plot(x1, y1)
plt.plot(x1, y2)
plt.plot(x1, y3)
plt.show()

# prophet 预测的结果都放在了变量 forecast 里面
print(forecast[['ds', 'yhat', 'yhat_lower', 'yhat_upper']])

Prophet 的参数设置

在 Prophet 中，用户一般可以设置以下四种参数:

Prophet 的默认参数配置(源码 forecaster.py):

class Prophet:

    def __init__(
        self,
        # 趋势项增长函数
        growth = 'linear',  # "logistic"
        # 变点
        changepoints = None,
        n_changepoints = 25,  # 变点的个数, 通过等分的方法找到 25 个变点
        changepoint_range = 0.8,  # 基于时间序列的前 80% 的历史数据中设置变点
        changepoint_prior_scale = 0.05,  # 变点增长率的分布情况，控制趋势的灵活度
        # 季节性
        seasonality_mode = 'additive',  # 季节性模式
        seasonality_prior_scale = 10.0,  # 季节性效应，控制季节项的灵活度。这个值越大，表示季节的效应越明显；这个值越小，表示季节的效应越不明显
        yearly_seasonality = 'auto',
        weekly_seasonality = 'auto',
        daily_seasonality = 'auto',
        # 节假日
        holidays = None,
        holidays_prior_scale = 10.0,  # 控制节假日的灵活度
        # 其它
        mcmc_samples = 0, # 贝叶斯估计 MCMC 采样
        interval_width = 0.80,  # 概率预测区间
        uncertainty_samples = 1000  # TODO
    ):
        pass

增长函数

在 Prophet 里面，有两个增长函数，分别是分段线性函数(linear)和逻辑回归函数(logistic). 而 m = Prophet() 默认使用的是分段线性函数(linear)，并且如果要是用逻辑回归函数的时候， 需要设置 capacity 的值，即 df[‘cap’] = 100，否则会出错。

m = Prophet()
m = Prophet(growth = 'linear')
m = Prophet(growth = 'logistic')

变点

在 Prophet 里面，变点默认的选择方法是前 80% 的点中等距选择 25 个点作为变点， 也可以通过以下方法来自行设置变点，甚至可以人为设置某些点。

m = Prophet(changepoints = ['2014-01-01'])
m = Prophet(n_changepoints = 25)
m = Prophet(changepoint_range = 0.8)
m = Prophet(changepoint_prior_scale = 0.05)

变点的作图:

from fbprophet.plot import add_changepoints_to_plot

m = Prophet()

future = model.make_future_dataframe(periods = 30, freq = "min")
forecast = model.predict(future)

fig = m.plot(forecast)
a = add_changepoints_to_plot(fig.gca(), m, forecast)

周期性

可以在 Prophet 中设置周期性，无论是按月(month)还是周(week)都是可以的。

model = Prophet(weekly_seasonality = False)
model.add_seasonality(name = "monthly", period = 30.5, fourier_order = 5)

model = Prophet(weekly_seasonality = True)
model.add_seasonality(name = "weekly", period = 7, fourier_order = 3, prior_scale = 0.1)

节假日

可以设置某些天是节假日，并且设置它的前后影响范围，也就是 lower_window 和 upper_window

# build holidays datasets
playoffs = pd.DataFrame({
    "holiday": "playoff",
    "ds": pd.to_datetime([
        '2008-01-13', '2009-01-03', '2010-01-16',
        '2010-01-24', '2010-02-07', '2011-01-08',
        '2013-01-12', '2014-01-12', '2014-01-19',
        '2014-02-02', '2015-01-11', '2016-01-17',
        '2016-01-24', '2016-02-07'
    ]),
    "lower_window": 0,
    "upper_window": 1,
})

superbowls = pd.DataFrame({
    "holiday": "superbowl",
    "ds": pd.to_datetime(['2010-02-07', '2014-02-02', '2016-02-07']),
    "lower_window": 0,
    "upper_window": 1,
})
holidays = pd.concat([playoffs, superbowl])

# model 
model = Prophet(holidays = holidays, holidays_prior_scale = 10.0)

对于商业分析等领域的时间序列，Prophet 可以进行很好的拟合和预测，但是对于一些周期性或者趋势性不是很强的时间序列， 用 Prophet 可能就不合适了。 但是，Prophet 提供了一种时序预测的方法，在用户不是很懂时间序列的前提下都可以使用这 个工具得到一个能接受的结果。 具体是否用 Prophet 则需要根据具体的时间序列来确定。

饱和预测

Staturating Forecast

Forecasting Growth

默认情况下，Prophet 使用线性模型进行预测。在预测增长时，通常有一些可达到的最大点： 市场总规模、总人口规模等。这称为承载能力，预测应在该点饱和。

import pandas as pd

# ============================================
# data
# ============================================
df = pd.read_csv("./data/example_wp_log_R.csv")
# 设定一个 carrying capacity,根据数据或专家经验说明市场规模
df["cap"] = 8.5

# ============================================
# model 1 - staturating maximum
# ============================================
m = Prophet(growth = "logistic")
m.fit(df)

future_df = m.make_future_dataframe(periods = 1826)
future_df["cap"] = 8.5
forecast = m.predict(future_df)
fig = m.plot(forecast)

Saturating Minimum

Saturating Minimum, 饱和最小值

逻辑增长模型还可以处理饱和最小值，该最小值是用列floor指定的，与cap列指定最大值的方式相同

# data
df = pd.read_csv("./data/example_wp_log_R.csv")
df["y"] = 10 - df["y"]
df["cap"] = 6
df["floor"] = 1.5

# ------------------------------
# staturating maximum and minimum
# ------------------------------
m = Prophet(growth = "logistic")
m.fit(df)

future_df = m.make_future_dataframe(periods = 1826)
future_df["cap"] = 6
future_df["floor"] = 1.5
forecast = m.predict(future_df)

fig = m.plot(forecast)

季节性和节假日效应和建模

Seasonality, Holiday Effects, And Regressors

假期和特殊事件建模

如果一个需要建模的时间序列中存在假期或其他重要的事件， 则必须为这些特殊事件创建单独的 DataFrame. Dataframe 的格式如下:

1.创建一个 DataFrame，其中包括了 Peyton Manning 所有季后赛出场的日期:

# data
df = pd.read_csv("./data/example_wp_log_peyton_manning.csv")

# 季后赛
playoffs = pd.DataFrame({
   "holiday": "playoff",
   "ds": pd.to_datetime(["2018-01-13", "2019-01-03", "2010-01-16",
                         "2010-01-24", "2010-02-07", "2011-01-08",
                         "2013-01-12", "2014-01-12", "2014-01-19",
                         "2014-02-02", "2015-01-11", "2016-01-17",
                         "2016-01-24", "2016-02-07"]),
   "lower_window": 0,
   "upper_window": 1
})

# 超级碗比赛
superbowls = pd.DataFrame({
   "holiday": "superbowl",
   "ds": pd.to_datetime(["2010-02-07", "2014-02-02", "2016-02-07"]),
   "lower_window": 0,
   "upper_window": 1
})
# 所有的特殊比赛
holiday = pd.concat([playoffs, superbowls])

print(holiday)

img

2.通过使用 holidays 参数传递节假日影响因素

from fbprophet import Prophet

m = Prophet(holidays = holidays)
m.fit(df)
future = m.make_future_dataframe(periods = 365)
forecast = m.predict(future)

forecast[(forecast["playoff"] + forecast["superbowl"]).abs() > 0][["ds", "playoff", "superbowl"]][-10:]

时间序列组件可视化:

prophet_plot_components(m, forecast)

plot_forecast_components(m, forecast, "superbowl")

指定内置的国家/地区假期

Build-in Country Holiday

# 模型-指定假期
m = Prophet(holidays = holidays)
m.add_country_holidays(country_name = "US")
m.fit(df)

# 所有指定的假期
m.train_holiday_names

forecast = m.predict(future)
fig = m.plot_components(forecast)

季节性的傅里叶变换

Fourier Order for Seasonalities

使用 偏傅里叶和(Partial Fourier Sum) 估计季节性，逼近非定期信号

from fbprophet.plot import plot_yearly

m = Prophet().fit(df)
a = plot_yearly(m)

通常季节性 yearly_seasonality 的默认值是比较合适的，但是当季节性需要适应更高频率的变化时，可以增加频率。 但是增加频率后的序列不会太平滑。 即可以在实例化模型时，可以为每个内置季节性指定傅里叶级别

from fbprophet.plot import plot_yearly

m = Prophet(yearly_seasonality = 20).fit(df)
a = plot_yearly(m)

增加傅里叶项的数量可以使季节性适应更快的变化周期，但也可能导致过拟合: N 傅里叶项对应于用于建模循环的 2N 变量。

指定自定义季节性

如果时间序列长度超过两个周期，Prophet 将默认自适应每周、每年的季节性。 它还适合时间序列的每日季节性，可以通过 add_seasonality 方法添加其他季节性，比如: 每月、每季度、每小时

m = Prophet(weekly_seasonality = False)
m.add_seasonality(name = "monthly", period = 30.5, fourier_order = 5)
forecast = m.fit(df).predict(future)
fig = m.plot_components(forecast)

参考