treats

ranking-agent · Jan 23, 2023 · f4c8ab8 · f4c8ab8
1 parent a66a02c
commit f4c8ab8
Show file tree

Hide file tree

Showing 4 changed files with 2,473 additions and 4,970 deletions.
diff --git a/src/rules/parse_to_trapi.py b/src/rules/parse_to_trapi.py
@@ -83,8 +83,9 @@ def parse_line(line, header, source, rule_conf, qmap):
     if float(rule["Std Confidence"]) <= minconf:
         return {}
     # Is the rule going to bring back too many results?
-    maxbody = source.get("max_body_size", 99999999999)
-    if int(rule["Body size"]) >= maxbody:
+    maxnegs = source.get("max_negative_examples", 99999999999)
+    negatives = int(rule["Body size"]) - int(rule["Positive Examples"])
+    if int(negatives) >= maxnegs:
         return {}
     template = rule_to_trapi(rule['Rule'],rule_conf,qmap)
     if template is not None:

diff --git a/src/rules/rule_config.json b/src/rules/rule_config.json
@@ -5,9 +5,9 @@
     "AMIE_inputs": [
       {
         "input_file": "robokop_biolink3_fixed_treats_len3_correct_rules.txt",
-        "min_positive_example": 2000,
-        "min_std_confidence": 0.001,
-        "max_body_size": 100000
+        "min_positive_example": 1000,
+        "min_std_confidence": 0.01,
+        "max_negative_examples": 10000
       }
     ],
     "source": "biolink:ChemicalEntity",