(pred: EvalPrediction)
| 51 | return np.count_nonzero(tokens != tokenizer.pad_token_id) |
| 52 | |
| 53 | def decode_pred(pred: EvalPrediction) -> Tuple[List[str], List[str]]: |
| 54 | return decode_tree_str(pred.predictions, tokenizer), decode_tree_str(pred.label_ids, tokenizer) |
| 55 | |
| 56 | def extraction_metrics(pred: EvalPrediction) -> Dict: |
| 57 | pred_str, label_str = decode_pred(pred) |
no test coverage detected