def extract_rolling_PSD(self, seriesname='raw', window_size=512): """Extracts power spectral density (PSD) for the specified series""" new_feature_name = seriesname+'_rolling_PSD_'+str(window_size) self.series[new_feature_name]=rolling_windows.rolling_power_ratio( self.series[seriesname], window_size=window_size )
except ImportError: print "Please create a dev_settings.py. Example: dev_settings.py.example" data = pd.read_table( join(SAVE_URL, 'raw', '20101214163931.a.rawwave_label.csv')) grouped = data.groupby(('taskid')) taskids = grouped.size().keys() ret = grouped.mean() # add new columns, to populate below ret['rolling_median'] = None ret['rolling_PSD'] = None for taskid in taskids: eeg_signal = grouped.get_group(taskid)[' Value'] # TODO remove space from name ret.set_value( taskid, 'rolling_median', rolling_windows.downsampled_rolling_median(eeg_signal, window_size=128)) window_size = 512 if len(eeg_signal) >= window_size: ret.set_value( taskid, 'rolling_PSD', rolling_windows.rolling_power_ratio(eeg_signal, window_size=512)) outfilename = 'features.pickle' ret.save(outfilename) print 'saved features as pickle in file "%s"' % outfilename
try: # Import config params from dev_settings import * except ImportError: print "Please create a dev_settings.py. Example: dev_settings.py.example" data = pd.read_table(join(SAVE_URL,'raw','20101214163931.a.rawwave_label.csv')) grouped = data.groupby(('taskid')) taskids = grouped.size().keys() ret = grouped.mean() # add new columns, to populate below ret['rolling_median'] = None ret['rolling_PSD'] = None for taskid in taskids: eeg_signal = grouped.get_group(taskid)[' Value'] # TODO remove space from name ret.set_value(taskid, 'rolling_median', rolling_windows.downsampled_rolling_median(eeg_signal, window_size=128)) window_size = 512 if len(eeg_signal) >= window_size: ret.set_value(taskid, 'rolling_PSD', rolling_windows.rolling_power_ratio(eeg_signal, window_size=512)) outfilename = 'features.pickle' ret.save(outfilename) print 'saved features as pickle in file "%s"' % outfilename