Ale37 commited on
Commit
05690d2
·
1 Parent(s): e11d622

results plot method

Browse files
Files changed (1) hide show
  1. 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