Added filter to remove stimuli with no fixations

matthias-k · Dec 16, 2023 · a4d8803 · a4d8803
1 parent b3ae536
commit a4d8803
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/pysaliency/filter_datasets.py b/pysaliency/filter_datasets.py
@@ -288,3 +288,10 @@ def filter_scanpaths_by_length(scanpaths: FixationTrains, intervals: list):
     scanpaths = scanpaths.filter_fixation_trains(indices)
 
     return scanpaths
+
+
+def remove_stimuli_without_fixations(stimuli: Stimuli, fixations: Fixations):
+    """Remove stimuli with no fixations"""
+
+    stimuli_indices_with_fixations = list(set(fixations.n))
+    return create_subset(stimuli, fixations, stimuli_indices_with_fixations)