diff options
Diffstat (limited to 'src/argaze/GazeAnalysis/Entropy.py')
-rw-r--r-- | src/argaze/GazeAnalysis/Entropy.py | 10 |
1 files changed, 5 insertions, 5 deletions
diff --git a/src/argaze/GazeAnalysis/Entropy.py b/src/argaze/GazeAnalysis/Entropy.py index a73901e..5bac43e 100644 --- a/src/argaze/GazeAnalysis/Entropy.py +++ b/src/argaze/GazeAnalysis/Entropy.py @@ -17,11 +17,11 @@ __credits__ = [] __copyright__ = "Copyright 2023, Ecole Nationale de l'Aviation Civile (ENAC)" __license__ = "GPLv3" +import numpy + from argaze import GazeFeatures, DataFeatures from argaze.GazeAnalysis import TransitionMatrix -import pandas -import numpy class AOIScanPathAnalyzer(GazeFeatures.AOIScanPathAnalyzer): """Implementation of entropy algorithm as described in: @@ -66,12 +66,12 @@ class AOIScanPathAnalyzer(GazeFeatures.AOIScanPathAnalyzer): scan_fixations_count, aoi_fixations_count = aoi_scan_path.fixations_count() # Probability to have a fixation onto each aoi - stationary_probalities = {aoi: count/scan_fixations_count for aoi, count in aoi_fixations_count.items()} + stationary_probabilities = {aoi: count/scan_fixations_count for aoi, count in aoi_fixations_count.items()} # Stationary entropy self.__stationary_entropy = 0 - for aoi, p in stationary_probalities.items(): + for aoi, p in stationary_probabilities.items(): self.__stationary_entropy += p * numpy.log(p + 1e-9) @@ -84,7 +84,7 @@ class AOIScanPathAnalyzer(GazeFeatures.AOIScanPathAnalyzer): for aoi, s in destination_p_log_sum.items(): - self.__transition_entropy += s * stationary_probalities[aoi] + self.__transition_entropy += s * stationary_probabilities[aoi] self.__transition_entropy *= -1 |