aurora

1
import numpy as np
2
from typing import Dict, Sequence, Tuple, Union
3

4

5
def compute_accuracy(eval_preds: Sequence[Union[np.ndarray, Tuple[np.ndarray]]]) -> Dict[str, float]:
6
    preds, _ = eval_preds
7
    return {"accuracy": (preds[0] > preds[1]).sum() / len(preds[0])}
8