Fixing corner case in local anomaly detectors trained with one row

bigmlcom · Jun 10, 2020 · 513b9d3 · 513b9d3
1 parent 9310f2a
commit 513b9d3
Show file tree

Hide file tree

Showing 4 changed files with 13 additions and 4 deletions.
diff --git a/HISTORY.rst b/HISTORY.rst
@@ -3,6 +3,12 @@
 History
 -------
 
+4.32.1 (2020-06-10)
+~~~~~~~~~~~~~~~~~~~
+
+- Fixing local anomaly scores prediction for corner cases of samples with
+  one row.
+
 4.32.0 (2020-05-19)
 ~~~~~~~~~~~~~~~~~~~
 

diff --git a/bigml/anomaly.py b/bigml/anomaly.py
@@ -96,8 +96,8 @@ def __init__(self, anomaly, api=None):
                                         "Score will"
                                         " not be available")
                     else:
-                        default_depth = (
-                            2 * (DEPTH_FACTOR + \
+                        default_depth = self.mean_depth if \
+                            self.sample_size == 1 else (2 * (DEPTH_FACTOR + \
                             math.log(self.sample_size - 1) - \
                             (float(self.sample_size - 1) / self.sample_size)))
                         self.expected_mean_depth = min(self.mean_depth,
@@ -128,7 +128,9 @@ def anomaly_score(self, input_data):
             value between 0 and 1.
 
         """
-
+        # corner case with only one record
+        if self.sample_size == 1:
+            return 1
         # Checks and cleans input_data leaving the fields used in the model
         input_data = self.filter_input_data(input_data)
 

diff --git a/bigml/version.py b/bigml/version.py
@@ -1 +1 @@
-__version__ = '4.32.0'
+__version__ = '4.32.1'
diff --git a/docs/index.rst b/docs/index.rst
@@ -1,6 +1,7 @@
 .. toctree::
    :hidden:
 
+   index
    ml_resources
    creating_resources
    101_model