Update pages/2_Energy Demand Forecasting.py

pages/2_Energy Demand Forecasting.py

@@ -355,10 +355,15 @@ elif option=='Prophet':
             </h2>
             </div>
             """, unsafe_allow_html=True)
-    y_train_true['y']=y_train_true['y'].astype('float')
-    # y_train_pred['yhat']=y_train_pred['yhat'].astype('float')
-    y_test_true['y']=y_test_true['y'].astype('float')
-    y_test_pred['yhat']=y_test_pred['yhat'].astype('float')
+    # y_train_true['y']=y_train_true['y'].astype('float')
+    # # y_train_pred['yhat']=y_train_pred['yhat'].astype('float')
+    # y_test_true['y']=y_test_true['y'].astype('float')
+    # y_test_pred['yhat']=y_test_pred['yhat'].astype('float')
+    y_train_true.loc[:, 'y'] = y_train_true['y'].astype('float')
+    # y_train_pred.loc[:, 'yhat'] = y_train_pred['yhat'].astype('float')
+    y_test_true.loc[:, 'y'] = y_test_true['y'].astype('float')
+    y_test_pred.loc[:, 'yhat'] = y_test_pred['yhat'].astype('float')
+
     
     fig = go.Figure([
     # go.Scatter(x=y_train_true['ds'],y=y_train_true['y'],name='Train Observed',line=dict(color='blue')),