Jiqing commited on
Commit
b733fa5
1 Parent(s): a888915

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
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 = {"acc": accuracy_score(labels, preds), "mcc": matthews_corrcoef(labels, preds)}
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):