|
735 | 735 | ACC_PATTERN = { |
736 | 736 | "acc": r"^(?:\{\"accuracy|accuracy)[\": ]*=?\s*([\d\.]+).*", |
737 | 737 | "AUC": r"^AUC=([\d\.]+).*", |
738 | | - "mAP": r".*'(?:mAP|Total)':\s*([\d\.]+)", |
| 738 | + "mAP": r".*(?:mAP=|Total:)\s*([\d.]+)", |
739 | 739 | "bleu": r"^BLEU\:\s*([\d\.]+).*", |
740 | 740 | "F1": r"^{[\"\']exact_match[\"\']\:\s*[\d\.]+,\s*[\"\']f1[\"\']\:\s*([\d\.]+)}", |
741 | 741 | "WER": r"Word Error Rate\:.*, accuracy=([0-9\.]+)%", |
742 | 742 | "DICE": r"Accuracy\:\s*mean\s*=\s*([\d\.]+).*", |
743 | | - "ROUGE1": r".*'rouge1':\s([\d.]+).*", |
744 | | - "ROUGE2": r".*'rouge2':\s([\d.]+).*", |
745 | | - "ROUGEL": r".*'rougeL':\s([\d.]+).*", |
| 743 | + "ROUGE1": r".*'rouge1':\s+'?([\d.]+)'?.*", |
| 744 | + "ROUGE2": r".*'rouge2':\s+'?([\d.]+)'?.*", |
| 745 | + "ROUGEL": r".*'rougeL':\s+'?([\d.]+)'?.*", |
746 | 746 | "ROUGELSUM": r".*'rougeLsum':\s([\d.]+).*", |
747 | 747 | "GEN_LEN": r".*'gen_len':\s([\d.]+).*", |
748 | 748 | "TOKENS_PER_SAMPLE": r".*'tokens_per_sample':\s([\d.]+).*", |
749 | | - "CLIP_SCORE": r".*'CLIP_SCORE':\s([\d.]+).*", |
750 | | - "FID_SCORE": r".*'FID_SCORE':\s([\d.]+).*", |
| 749 | + "CLIP_SCORE": r".*'CLIP_SCORE':\s.([\d.]+).*", |
| 750 | + "FID_SCORE": r".*'FID_SCORE':\s.([\d.]+).*", |
751 | 751 | "gsm8k_accuracy": r".*'gsm8k':\s([\d.]+).*", |
752 | 752 | "mbxp_accuracy": r".*'mbxp':\s([\d.]+).*", |
753 | 753 | "exact_match": r".*'exact_match':\s([\d.]+).*" |
|
0 commit comments