File size: 516 Bytes
d758c99 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 | from typing import Dict, NamedTuple, Optional
import numpy as np
class EvalPredictionWithSize(NamedTuple):
"""
Evaluation output (always contains labels), to be used
to compute metrics.
"""
predictions: np.ndarray
predictions_size: np.ndarray
label_ids: np.ndarray
label_size: np.ndarray
class PredictionOutputWithSize(NamedTuple):
predictions: np.ndarray
predictions_size: np.ndarray
label_ids: Optional[np.ndarray]
label_size: Optional[np.ndarray]
metrics: Optional[Dict[str, float]] |