问题描述
用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