24 lines
878 B
Python
24 lines
878 B
Python
|
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()
|