feat: fixed entropy calculation

lab-v2 · Sep 29, 2023 · 33019eb · 33019eb
1 parent 6178d2a
commit 33019eb
Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/langdiversity/measures/shannon_entropy.py b/langdiversity/measures/shannon_entropy.py
@@ -8,4 +8,4 @@ class ShannonEntropyMeasure(AbstractMeasure):
     def generate(self, values: List[Hashable]) -> float:
         probabilities = empirical_probability(values=values)
         entropy = -sum(probability * log2(probability) for probability in probabilities.values() if probability > 0)
-        return entropy
+        return abs(entropy)
diff --git a/setup.py b/setup.py
@@ -8,7 +8,7 @@
 setup(
     name='langdiversity',
     packages=find_packages(exclude=['tests']),
-    version='1.1.1',
+    version='1.1.2',
     description='A tool to elevate your language models with insightful diversity metrics.',
     long_description=long_description,  
     long_description_content_type="text/markdown",