Fix input format

relation_extraction.py

@@ -72,8 +72,20 @@ class relation_extraction(evaluate.Metric):
             inputs_description=_KWARGS_DESCRIPTION,
             # This defines the format of each prediction and reference
             features=datasets.Features({
-                'predictions': datasets.Value('int64'),
-                'references': datasets.Value('int64'),
+                'predictions': datasets.Sequence(datasets.Sequence(datasets.Features({
+                    "head": datasets.Value("string"),
+                    "head_type": datasets.Value("string"),
+                    "type": datasets.Value("string"),
+                    "tail": datasets.Value("string"),
+                    "tail_type": datasets.Value("string"),
+                }))),
+                'references': datasets.Sequence(datasets.Sequence(datasets.Features({
+                    "head": datasets.Value("string"),
+                    "head_type": datasets.Value("string"),
+                    "type": datasets.Value("string"),
+                    "tail": datasets.Value("string"),
+                    "tail_type": datasets.Value("string"),
+                }))),
             }),
             # Homepage of the module for documentation
             homepage="http://module.homepage",