Example #1
0
 def _info(self):
     return MetricInfo(
         description="dummy metric for tests",
         citation="insert citation here",
         features=Features({
             "predictions": Value("int64"),
             "references": Value("int64")
         }),
     )
 def _info(self):
     return MetricInfo(
         description="dummy metric for tests",
         citation="insert citation here",
         features=Features({
             "inputs": Value("int64"),
             "targets": Value("int64")
         }),
     )
Example #3
0
 def _info(self):
     return MetricInfo(
         description="dummy metric for tests",
         citation="insert citation here",
         features=Features(
             {"predictions": Sequence(Value("int64")), "references": Sequence(Value("int64"))}
             if self.config_name == "multilabel"
             else {"predictions": Value("int64"), "references": Value("int64")}
         ),
     )