blob: f77b17b2408e7e750af4dd8b74aa887665d75942 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
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()
|
