eagle0504 commited on
Commit
a26a34d
1 Parent(s): 77fb945

Update utils/helper.py

Browse files
Files changed (1) hide show
  1. utils/helper.py +1 -1
utils/helper.py CHANGED
@@ -123,7 +123,7 @@ def create_datasets(data_scaled, look_back=50):
123
  return X, y
124
 
125
 
126
- def plot_predictions(train_data, test_data, y_train_pred, y_test_pred, model_name):
127
  fig = go.Figure()
128
  fig.add_trace(go.Scatter(x=train_data.index, y=train_data.values.flatten(), mode='lines', name='Training Data'))
129
  fig.add_trace(go.Scatter(x=test_data.index, y=test_data.values.flatten(), mode='lines', name='Test Data'))
 
123
  return X, y
124
 
125
 
126
+ def plot_predictions(train_data, test_data, y_train_pred, y_test_pred, model_name, look_back=50):
127
  fig = go.Figure()
128
  fig.add_trace(go.Scatter(x=train_data.index, y=train_data.values.flatten(), mode='lines', name='Training Data'))
129
  fig.add_trace(go.Scatter(x=test_data.index, y=test_data.values.flatten(), mode='lines', name='Test Data'))