Update test_onnx.py

clulab · Dec 16, 2021 · 82ba700 · bethard · Dec 16, 2021 · 82ba700
1 parent 9007983
commit 82ba700
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/main/src/main/python/test_onnx.py b/main/src/main/python/test_onnx.py
@@ -2,6 +2,7 @@
 import json
 import numpy as np
 from pytorch.seqScorer import *
+import time
 
 if __name__ == '__main__':
 
@@ -34,7 +35,7 @@
     ort_char = onnxruntime.InferenceSession("char.onnx")
 
     scoreCountsByLabel = ScoreCountsByLabel()
-
+    start_time = time.time()
     for taskId in range(0, taskManager.taskCount):
         taskName = taskManager.tasks[taskId].taskName
         sentences = taskManager.tasks[taskId].testSentences
@@ -78,5 +79,6 @@
     print (f"Micro F1 : {scoreCountsByLabel.f1()}")
     for label in scoreCountsByLabel.labels():
         print (f"\tP/R/F1 for label {label} ({scoreCountsByLabel.map[label].gold}): {scoreCountsByLabel.precision(label)} / {scoreCountsByLabel.recall(label)} / {scoreCountsByLabel.f1(label)}")
-
+    duration = time.time() - start_time
+    print (duration)