Bugfix: ExpSaliencyMapModel doesn't apply to probabilistic models (#47)

Signed-off-by: Matthias Kümmmerer <[email protected]>
matthias-k · Jan 27, 2024 · 1a6d965 · 1a6d965
1 parent 1b2e591
commit 1a6d965
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pysaliency/models.py b/pysaliency/models.py
@@ -12,7 +12,7 @@
 
 from .saliency_map_models import (SaliencyMapModel, ScanpathSaliencyMapModel, handle_stimulus,
                                   SubjectDependentSaliencyMapModel,
-                                  ExpSaliencyMapModel,
+                                  DensitySaliencyMapModel,
                                   DisjointUnionMixin,
                                   GaussianSaliencyMapModel,
                                   )
@@ -578,7 +578,7 @@ def get_saliency_map_model_for_sAUC(self, baseline_model):
 
     def get_saliency_map_model_for_NSS(self):
         return SubjectDependentSaliencyMapModel({
-            s: ExpSaliencyMapModel(self.subject_models[s])
+            s: DensitySaliencyMapModel(self.subject_models[s])
             for s in self.subject_models})