results plot method
Browse files- src/model.py +8 -0
src/model.py
CHANGED
|
@@ -4,6 +4,7 @@ from darts.metrics import mape
|
|
| 4 |
from darts.model import ExponentialSmoothing
|
| 5 |
from darts.utils.missing_values import fill_missing_values
|
| 6 |
from darts.dataprocessing.transformers import Scaler
|
|
|
|
| 7 |
import numpy as np
|
| 8 |
import pandas as pd
|
| 9 |
|
|
@@ -84,4 +85,11 @@ def inverse_scale_predictions(scaled_preds, scaler):
|
|
| 84 |
return preds
|
| 85 |
|
| 86 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 87 |
|
|
|
|
| 4 |
from darts.model import ExponentialSmoothing
|
| 5 |
from darts.utils.missing_values import fill_missing_values
|
| 6 |
from darts.dataprocessing.transformers import Scaler
|
| 7 |
+
import matplotlib.pyplot as plt
|
| 8 |
import numpy as np
|
| 9 |
import pandas as pd
|
| 10 |
|
|
|
|
| 85 |
return preds
|
| 86 |
|
| 87 |
|
| 88 |
+
def plot_results(train_ili, val_ili, exp_preds):
|
| 89 |
+
train_ili.plot(label='train')
|
| 90 |
+
val_ili.plot(label='val')
|
| 91 |
+
exp_preds.plot(label='preds')
|
| 92 |
+
plt.savefig('./charts/results.png')
|
| 93 |
+
|
| 94 |
+
|
| 95 |
|