Update README.md
Browse files
README.md
CHANGED
@@ -90,7 +90,7 @@ def create_scheduler(training_args, optimizer):
|
|
90 |
def compute_metrics(eval_preds):
|
91 |
probs, labels = eval_preds
|
92 |
preds = np.argmax(probs, axis=-1)
|
93 |
-
result = {"
|
94 |
return result
|
95 |
|
96 |
def preprocess_logits_for_metrics(logits, labels):
|
|
|
90 |
def compute_metrics(eval_preds):
|
91 |
probs, labels = eval_preds
|
92 |
preds = np.argmax(probs, axis=-1)
|
93 |
+
result = {"accuracy": accuracy_score(labels, preds), "mcc": matthews_corrcoef(labels, preds)}
|
94 |
return result
|
95 |
|
96 |
def preprocess_logits_for_metrics(logits, labels):
|