Diff of /chexbert/run_chexbert.py [000000] .. [4abb48]

Switch to side-by-side view

--- a
+++ b/chexbert/run_chexbert.py
@@ -0,0 +1,26 @@
+import subprocess
+from local_config import CHEXBERT_ENV_PATH, CHEXBERT_PATH
+
+
+def run_chexbert_labeler(reports_path, output_path):
+    # Path to a Python interpreter that runs any Python script
+    # under the virtualenv /path/to/virtualenv/
+    python_bin = CHEXBERT_ENV_PATH
+
+    # Path to the script that must run under the virtualenv
+    script_file = "label"
+
+    print("Starting to extract Chexbert labels...")
+
+    process = subprocess.run([python_bin, "-m", script_file, "-c", "checkpoint/chexbert.pth", "-d", reports_path, "-o", output_path],
+                             stdout=subprocess.PIPE, stderr=subprocess.PIPE, cwd=CHEXBERT_PATH)
+
+    # Print the output
+    print(process.stdout.decode())
+
+    # Print the errors if there are any
+    if process.stderr:
+        print("Errors:")
+        print(process.stderr.decode())
+
+    print("Finished extracting Chexbert labels.")