fix: machine learning only take results with > 90% confidence (#1875)

2023-02-26 04:02:35 +07:00 · 2023-02-26 04:02:35 +07:00 · 3d468c369c
parent 6c7679714b
commit 3d468c369c
1 changed files with 2 additions and 5 deletions
--- a/machine-learning/src/main.py
+++ b/machine-learning/src/main.py
@ -45,11 +45,8 @@ def run_engine(engine, path):

    for index, pred in enumerate(predictions):
        tags = pred['label'].split(', ')
-        if (index == 0):
-            result = tags
-        else:
-            if (pred['score'] > 0.5):
-                result = [*result, *tags]
+        if (pred['score'] > 0.9):
+            result = [*result, *tags]

    if (len(result) > 1):
        result = list(set(result))