Update utils/helper.py
Browse files- 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'))
|