Update pages/1_Store Demand Forecasting.py

pages/1_Store Demand Forecasting.py

@@ -266,6 +266,7 @@ if option=='TFT':
         train_a=train_dataset.loc[(train_dataset['store']==store) & (train_dataset['item']==item)][['date','store','item','sales']]
         test_a=test_dataset.loc[(test_dataset['store']==store) & (test_dataset['item']==item)][['date','store','item','sales']]
         actual_final_data=pd.concat([train_a,test_a])
+        actual_final_data['date']=actual_final_data['date'].dt.date
         tab3.dataframe(actual_final_data,width=500)
         
     except:
@@ -476,15 +477,16 @@ elif option=='Prophet':
                             )
 
     #------------------------------------------Tab-3--------------------------------------------------
-        train_a=fb_train_data.loc[fb_train_data[item]==1][['date','y']]
+        train_a=fb_train_data.loc[fb_train_data[item]==1][['ds','y']]
         train_a['store']=1
         train_a['item']=item
         train_a.rename({'y':'sales'})
-        test_a=fb_test_data.loc[fb_test_data[item]==1][['date','y']]
+        test_a=fb_test_data.loc[fb_test_data[item]==1][['ds','y']]
         test_a['store']=1
         test_a['item']=item
-        test_a.rename({'y':'sales'})
+        test_a.rename({'y':'sales',"ds":'date'})
         actual_final_data=pd.concat([train_a,test_a])
+        actual_final_data['ds']=actual_final_data['ds'].dt.date
         tab3.dataframe(actual_final_data,width=500)