Ale37 commited on
Commit
1847ddd
·
1 Parent(s): 3eef9a0

train exponential smoothing method

Browse files
Files changed (1) hide show
  1. src/model.py +13 -1
src/model.py CHANGED
@@ -65,7 +65,19 @@ def scale_train(train_ili):
65
  scaler = Scaler()
66
  train_ili_scaled = scaler.fit_transform(train_ili)
67
 
68
- return train_ili_scaled
 
 
 
 
 
 
 
 
 
 
 
 
69
 
70
 
71
 
 
65
  scaler = Scaler()
66
  train_ili_scaled = scaler.fit_transform(train_ili)
67
 
68
+ return train_ili_scaled, scaler
69
+
70
+
71
+ def train_exponential_smoothing(train_ili_scaled, val_ili):
72
+ exp_sm = ExponentialSmoothing()
73
+ exp_sm.fit(train_ili_scaled)
74
+ exp_preds = exp_sm.predict(len(val_ili))
75
+
76
+ # print('Exponential Smoothing mape:', mape(val_ili, scaler.inverse_transform(exp_preds)))
77
+
78
+ return exp_preds
79
+
80
+
81
 
82
 
83