时间序列分析-Prophet 模块中使用 logistic 增长趋势模型进行预测时报错问题

问题描述

用prophet进行预测时报错：ValueError: Capacities must be supplied for logistic growth in column “cap”。

问题说明

用prophet做时序数据预测时，growth的方式选择logistic，该方式同时要指定承载能力（当预测增长情况时，通常会存在可到达的最大极限值，例如：总市场规模、总人口数等等。这被称做承载能力（carrying capacity），那么预测时就应当在接近该值时趋于饱和）。
如果只在prophet函数中指定growth=‘logistic’,预测结果中就会报错没有‘cap’列。

解决方式

方式1、在数据进入模型之前，除了有[‘ds’,‘y’]还需要增加一列‘cap’，且在构建待预测日期数据框后，需要给该dataframe增加一列‘cap’。代码如下图所示：

本例中我选择的承载能力值为60，读者根据自己的数据情况修改。

方式2、修改make_future_dataframe函数。
分析源代码，发现在构造训练数据或预测数据时，有提到cap，如下图所示

我们选择模型为Prophet(growth=‘logistic’)时，构造的数据要有[‘ds’, ‘y’, ‘cap’]列，所以我们必须提供含有cap列的dataframe，并且在构建待预测日期数据框时，也要把cap列保留，但是源码中该函数缺少cap列，源码如下图所示：

至于如何修改，可以在此函数上修改，也可以新增一个函数来实现此功能。
prophet的具体使用方式可参考：https://zhuanlan.zhihu.com/p/636279576