如何获取FB Prophet的特征重要性?

4
我正在使用FB Prophet进行时间序列预测。我添加了两个特征——折扣和促销,并加入了节假日效应。模型拟合得很好。但我想获得特征重要性,以检查这两个特征的贡献有多大。似乎FB Prophet没有像其他机器学习模型“model.feature_importances_”那样的特征重要性函数。
在FB Prophet中,我可以获得“forecast”数据框,其中包含: 趋势 yhat_lower yhat_upper trend_lower trend_upper discount_x discount_lower discount_upper extra_regressors_multiplicative extra_regressors_multiplicative_lower extra_regressors_multiplicative_upper holidays holidays_lower holidays_upper multiplicative_terms multiplicative_terms_lower multiplicative_terms_upper promotion_x promotion_lower promotion_upper promotion_Day promotion_Day_lower promotion_Day_upper weekly weekly_lower weekly_upper additive_terms additive_terms_lower additive_terms_upper yhat y
在这种情况下,我如何分析特征重要性?
2个回答

2
你可以使用以下代码获取系数:
from prophet.utilities import regressor_coefficients
regressor_coef = regressor_coefficients(model)
regressor_coef[['regressor', 'regressor_mode', 'coef']].sort_values('coef')*

0
使用SHAP来获取它。它会告诉你每个特征对于模型的贡献程度以及其重要性。

网页内容由stack overflow 提供, 点击上面的
可以查看英文原文,
原文链接