Feat: add method that change threshold

asheswook · May 29, 2023 · 808c27e · 808c27e
1 parent e3792d0
commit 808c27e
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/VisageSnap/main.py b/VisageSnap/main.py
@@ -23,6 +23,15 @@ def __init__(self, globalState: GlobalState = None, directory: Directory = None)
     def set_directory(self, directory: Directory) -> None:
         self.__directory = directory
 
+    @property
+    def threshold(self) -> float:
+        return self.predictor.threshold
+
+    @threshold.setter
+    def threshold(self, value: float) -> None:
+        assert isinstance(value, float), "threshold must be float."
+        self.predictor.threshold = value
+
     def load_model(self) -> None:
         """
         This function loads the model.