Spaces:

Intel
/

bias_auc

Running

tybrs commited on Jun 28, 2024

Commit

eb51dce

verified ·

1 Parent(s): 5254f6a

Update bias_auc.py

changed label input format

Files changed (1) hide show

bias_auc.py CHANGED Viewed

@@ -67,7 +67,7 @@ class BiasAUC(evaluate.Metric):
             features=datasets.Features(
                 {
                  'target': Sequence(feature=Value(dtype='string', id=None), length=-1, id=None),
-                 'label': Value(dtype='int64', id=None),
                  'output': Sequence(feature=Value(dtype='float32', id=None), length=-1, id=None),
                 }
             ),

             features=datasets.Features(
                 {
                  'target': Sequence(feature=Value(dtype='string', id=None), length=-1, id=None),
+                 'label': Sequence(feature=Value(dtype='int64', id=None), length=-1, id=None),
                  'output': Sequence(feature=Value(dtype='float32', id=None), length=-1, id=None),
                 }
             ),