added metrics

my_metrics.py

@@ -0,0 +1,7 @@
+import sklearn.metrics
+import numpy as np
+
+def f1_macro(targets, predictions):
+      targets, predictions = np.asarray(targets).astype(str), np.asarray(predictions).astype(str)
+      return {"f1_macro": 100*sklearn.metrics.f1_score(targets, predictions, average='macro')}
+

tasks.py

@@ -4,6 +4,7 @@ import functools
 import seqio
 import tensorflow_datasets as tfds
 from t5.evaluation import metrics
+import mymetrics
 from t5.data import preprocessors
 import t5
 import tensorflow.compat.v1 as tf
@@ -59,7 +60,7 @@ seqio.TaskRegistry.add(
       categorise_preprocessor,
       seqio.preprocessors.tokenize_and_append_eos,
     ],
-    #metric_fns=[metrics.bleu],
+    metric_fns=[metrics.accuracy,my_metrics.f1_macro,metrics.bleu],
     output_features=DEFAULT_OUTPUT_FEATURES,
 )