Georg Willer
commited on
Commit
·
b9f7415
1
Parent(s):
e76f24f
Change default header names to upercase starting & add checking for alternative time_header names
Browse files- eyetrack2saccade.py +1 -1
- eyetrack_2_saccade_pipeline.py +1 -1
eyetrack2saccade.py
CHANGED
@@ -26,7 +26,7 @@ class Eye2SacExtractor:
|
|
26 |
def _map_relevant_data(self, time_header: str, x_headers: Union[str, list], y_headers: Union[str, list]):
|
27 |
# map and extract relevant data
|
28 |
try:
|
29 |
-
self.time = self.
|
30 |
if isinstance(x_headers, str):
|
31 |
self.x = self._get_value_array(x_headers, X_ALIASES)
|
32 |
# if the user accidently specified the header as list with a single entry, capture it
|
|
|
26 |
def _map_relevant_data(self, time_header: str, x_headers: Union[str, list], y_headers: Union[str, list]):
|
27 |
# map and extract relevant data
|
28 |
try:
|
29 |
+
self.time = self._get_value_array(time_header, TIME_ALIASES)
|
30 |
if isinstance(x_headers, str):
|
31 |
self.x = self._get_value_array(x_headers, X_ALIASES)
|
32 |
# if the user accidently specified the header as list with a single entry, capture it
|
eyetrack_2_saccade_pipeline.py
CHANGED
@@ -23,7 +23,7 @@ class Eye2SacPipeline(Pipeline):
|
|
23 |
preprocess_kwargs["y_headers"] = kwargs["y_headers"]
|
24 |
return preprocess_kwargs, {}, {}
|
25 |
|
26 |
-
def preprocess(self, inputs, missing=0.0, minlen=5, maxvel=40, maxacc=30, time_header = '
|
27 |
return self.eye2SacExtractor.extract_features(inputs, time_header, x_headers, y_headers, missing, minlen, maxvel, maxacc)
|
28 |
|
29 |
def _forward(self, model_inputs):
|
|
|
23 |
preprocess_kwargs["y_headers"] = kwargs["y_headers"]
|
24 |
return preprocess_kwargs, {}, {}
|
25 |
|
26 |
+
def preprocess(self, inputs, missing=0.0, minlen=5, maxvel=40, maxacc=30, time_header = 'Time', x_headers = 'X', y_headers= 'Y'):
|
27 |
return self.eye2SacExtractor.extract_features(inputs, time_header, x_headers, y_headers, missing, minlen, maxvel, maxacc)
|
28 |
|
29 |
def _forward(self, model_inputs):
|