Merge pull request #76 from RUTILEA/feature/set_auto_sigma_threshold

Update learning_model.py

src/main/python/model/learning_model.py

@@ -6,7 +6,6 @@
 from statistics import stdev, mean
 from math import sqrt
 
-
 class TestResults(object):
     def __init__(self):
         self.__distances_of_train_images = np.empty(shape=0)
@@ -198,7 +197,7 @@ def test(self, predict_training=False):
             else:
                 self.test_results.reload(distances_of_ok_images=pred_of_ok_images, distances_of_ng_images=pred_of_ng_images)
             if self.test_results.distances_of_ng_images.size != 0:
-                self.threshold = max(self.test_results.distances_of_ng_images)  # default threshold FIXME: logic
+                self.threshold = max(self.test_results.distances_of_ng_images.max(), np.percentile(self.test_results.distances_of_ok_images, 0.13))  # default threshold is the larger of max NG distance and 0.13 percentile (-3 sigma) of OK distances
                 self.__should_test = False
         except IndexError:  # TODO: handle as UndoneTrainingError
             print('TODO: tell the user to train')