(self, latest_metrics, type_path)
| 150 | return {"log": metrics, "preds": preds, f"{prefix}_loss": loss, f"{prefix}_{self.val_metric}": rouge_tensor} |
| 151 | |
| 152 | def save_metrics(self, latest_metrics, type_path) -> None: |
| 153 | self.metrics[type_path].append(latest_metrics) |
| 154 | save_json(self.metrics, self.metrics_save_path) |
| 155 | |
| 156 | def calc_generative_metrics(self, preds, target) -> Dict: |
| 157 | return calculate_rouge(preds, target) |
no test coverage detected