diff --git a/models/nerulforcastmodels.py b/models/nerulforcastmodels.py index 4ac6861..134d920 100644 --- a/models/nerulforcastmodels.py +++ b/models/nerulforcastmodels.py @@ -196,6 +196,8 @@ def ex_Model(df,horizon,input_size,train_steps,val_check_steps,early_stop_patien #进行未来时间预测 df_predict=nf.predict(df_test).reset_index() + # 去掉index列 + df_predict.drop(columns=['index'], inplace=True) df_predict.astype({col: 'float32' for col in df_predict.columns if col not in ['ds'] }) # 添加预测时间