Victoria Oberascher commited on
Commit
490df31
·
1 Parent(s): 6d18e2a

fix bug concerning input format

Browse files
Files changed (1) hide show
  1. horizon-metrics.py +6 -2
horizon-metrics.py CHANGED
@@ -138,9 +138,13 @@ class HorizonMetrics(evaluate.Metric):
138
  # This defines the format of each prediction and reference
139
  features=datasets.Features({
140
  'predictions':
141
- datasets.Sequence(datasets.Value("float")),
 
 
142
  'references':
143
- datasets.Sequence(datasets.Value("float")),
 
 
144
  }),
145
  codebase_urls=["http://github.com/path/to/codebase/of/new_module"])
146
 
 
138
  # This defines the format of each prediction and reference
139
  features=datasets.Features({
140
  'predictions':
141
+ datasets.Sequence(
142
+ datasets.Sequence(
143
+ datasets.Sequence(datasets.Value("float")))),
144
  'references':
145
+ datasets.Sequence(
146
+ datasets.Sequence(
147
+ datasets.Sequence(datasets.Value("float")))),
148
  }),
149
  codebase_urls=["http://github.com/path/to/codebase/of/new_module"])
150