train exponential smoothing method
Browse files- 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 |
|