Skip to content

Commit bc86819

Browse files
committed
fixes #10
1 parent c0974c3 commit bc86819

1 file changed

Lines changed: 5 additions & 1 deletion

File tree

src/edgeml/predictor.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -58,8 +58,12 @@ def predict(self):
5858
features = tsfresh.extract_features(
5959
window, column_id="id", default_fc_parameters=settings, n_jobs=0, disable_progressbar=True
6060
)
61-
6261
l = features.iloc[0].values.tolist()
62+
63+
if self.scaler is not None:
64+
for i in range(len(l)):
65+
l[i] = (l[i] - self.scaler["center"][i]) / self.scaler["scale"][i]
66+
6367
pred = self.predictor(l)
6468

6569
return {

0 commit comments

Comments
 (0)