Switch to side-by-side view

--- a
+++ b/scripts/plcom2012/evaluate.py
@@ -0,0 +1,39 @@
+import pickle
+import os
+os.environ['TF_CPP_MIN_LOG_LEVEL'] = "3"
+from os.path import dirname, realpath
+import sys
+sys.path.append(dirname(dirname(dirname(realpath(__file__)))))
+
+from sybil.parsing import parse_args
+from scripts.plcom2012.plcom2012 import PLCOm2012
+from sybil.utils.helpers import get_dataset
+import sybil.utils.loading as loaders
+
+def main(args):
+    # Load dataset and add dataset specific information to args
+    print("\nLoading data...")
+    test_data = loaders.get_eval_dataset_loader(
+            args,
+            get_dataset(args.dataset, 'test', args),
+            False
+            )
+    
+    model = PLCOm2012(args)
+
+    print("\nParameters:")
+    for attr, value in sorted(args.__dict__.items()):
+        if attr not in ['optimizer_state', 'patient_to_partition_dict', 'path_to_hidden_dict', 'exam_to_year_dict', 'exam_to_device_dict', 'treatment_to_index','drug_to_y']:
+            print("\t{}={}".format(attr.upper(), value))
+
+    print("-------------\nTesting on PLCOm2012")
+    model.save_prefix = 'test_'
+    model.test(test_data)
+    
+    print("Saving args to {}".format(args.results_path))
+    pickle.dump(vars(args), open(args.results_path,'wb'))
+
+if __name__ == '__main__':
+    __spec__ = "ModuleSpec(name='builtins', loader=<class '_frozen_importlib.BuiltinImporter'>)"
+    args = parse_args()
+    main(args)