для цикла данных Python fbprophet ошибка индекса кадра данных при прогнозировании временных рядов нескольких элементов

У меня похожая проблема, как и в этом вопросе: прогнозирование значений для каждой категории с использованием Prophet в python Я хочу сделать прогноз для несколько элементов на основе ключа pcodeid, поэтому для каждого pcodeid мне нужен отдельный прогноз.

Я изменил свой код, чтобы он соответствовал коду из лучшего ответа по ссылке выше, но я продолжаю получать сообщение об ошибке.

Часть моих данных (весь набор данных - это месячные данные за 4 года и за 50 кодов):

DS Pcodeid Y
2015-01-01 Австралия-P8205 10.36
2015-01-01 Австралия-P8206 2.06
2015-01-01 Австралия-P8208 7.1
2015-01-01 Австралия-P8211 39,76
2015-01-01 Австралия P8212 0,61
2015-01-01 Австралия-P8220 10.38
2015-01-01 Австралия-P8223 10.35
2015-01-01 Австралия-P8227 2.99 < br> 01.01.2015  АВСТРАЛИЯ-P8228  2.99
01.01.2015  АВСТРАЛИЯ-P8233  0.28
01.01.2015  АВСТРАЛИЯ-P8238  4544
01.01.2015  2015-01-01  АВСТРАЛИЯ-P8247  7,53

мой код:

def get_prediction(df):
    prediction = {}
    df2 = df.rename(columns={'PcodeID' : 'pcodeid'})
    list_pcodeid = df2.pcodeid.unique()
    
    for pcodeid in list_pcodeid:
        pcodeid_df = df2.loc[df2['pcodeid'] == pcodeid]
        # set the uncertainty interval to 95% (the Prophet default is 80%)
        my_model = Prophet()
        my_model.fit(pcodeid_df)
        future_dates = my_model.make_future_dataframe(periods=12, freq='MS')
        prediction = my_model.predict(future_dates)
        prediction[pcodeid] = prediction
    return prediction

Я продолжаю получать сообщение об ошибке для строки 7 pcodeid не определен. Парень из решения по ссылке прокомментировал, что у него это сработало отлично. Как мне определить pcodeid для работы, как описано в вопросе по ссылке?


person woliwia    schedule 27.11.2019    source источник