Victoria Oberascher commited on
Commit
7256d72
·
1 Parent(s): 30abd8c
Files changed (1) hide show
  1. horizon-metrics.py +2 -0
horizon-metrics.py CHANGED
@@ -180,6 +180,8 @@ class HorizonMetrics(evaluate.Metric):
180
  try:
181
  print("Predicitons:", predictions)
182
  print("Reference:", references)
 
 
183
  except:
184
  pass
185
 
 
180
  try:
181
  print("Predicitons:", predictions)
182
  print("Reference:", references)
183
+ self.ground_truth_det = references[0]
184
+ self.predictions = predictions[0]
185
  except:
186
  pass
187