Update utils.py

utils.py

@@ -201,6 +201,20 @@ def compute_metrics(eval_pred):
     #Before passing your predictions to compute, you need to convert the predictions to logits (remember all Transformers models return logits):
     return metric.compute(predictions=predictions, references=labels)
 
+
+def compute_metrics2(p):
+    pred, labels = p
+    pred = np.argmax(pred, axis=1)
+
+    accuracy = accuracy_score(y_true=labels, y_pred=pred)
+    recall = recall_score(y_true=labels, y_pred=pred)
+    precision = precision_score(y_true=labels, y_pred=pred)
+    f1 = f1_score(y_true=labels, y_pred=pred)
+
+    return {"accuracy": accuracy, "precision": precision, "recall": recall, "f1": f1}
+
+
+
                 
 def convert_to_markdown(text):
     text = text.replace("$","$")