abc

pipeline/tsfresh_basic.py

@@ -0,0 +1,23 @@
+from tsfresh.feature_extraction import extract_features, MinimalFCParameters
+from tsfresh.utilities.dataframe_functions import impute
+
+from utils.util import *
+
+
+class TSFreshBasicExtractor:
+    def __init__(self):
+        #print "Feature extraction: tsfresh basic"
+        self.extracted_features = []
+
+    def fit(self, x, y=None):
+        return self
+
+    def transform(self, df):
+        df_stacked = convert(df)
+        extracted_features = extract_features(df_stacked,
+                                              column_id="id",
+                                              column_kind="kind",
+                                              column_value="value",
+                                              default_fc_parameters=MinimalFCParameters())
+        self.extracted_features = impute(extracted_features)
+        return self.extracted_features.values.tolist()