Mark Color Modification
This commit is contained in:
parent
9bbfcaf736
commit
cdbb1de972
2 changed files with 5 additions and 22 deletions
|
|
@ -35,8 +35,8 @@ def make_table(result_dict):
|
||||||
"""Generate table of results."""
|
"""Generate table of results."""
|
||||||
md_writer = MarkdownTableWriter()
|
md_writer = MarkdownTableWriter()
|
||||||
latex_writer = LatexTableWriter()
|
latex_writer = LatexTableWriter()
|
||||||
md_writer.headers = ["Model", "Precision", "ppl_result"]#["Model", "Precision", "en", "zh"]
|
md_writer.headers = ["Model", "Precision", "ppl_result"]
|
||||||
latex_writer.headers = ["Model", "Precision", "ppl_result"]#"en", "zh"]
|
latex_writer.headers = ["Model", "Precision", "ppl_result"]
|
||||||
|
|
||||||
languages = ["en", "zh"]
|
languages = ["en", "zh"]
|
||||||
values = []
|
values = []
|
||||||
|
|
@ -46,7 +46,7 @@ def make_table(result_dict):
|
||||||
for language in languages:
|
for language in languages:
|
||||||
task_results = prec_results.get(language, None)
|
task_results = prec_results.get(language, None)
|
||||||
if task_results is None:
|
if task_results is None:
|
||||||
continue#value.append("")
|
continue
|
||||||
else:
|
else:
|
||||||
result = task_results["results"]
|
result = task_results["results"]
|
||||||
value.append("%.4f" % result)
|
value.append("%.4f" % result)
|
||||||
|
|
|
||||||
|
|
@ -106,9 +106,8 @@ def main():
|
||||||
|
|
||||||
# Add display of FP16 values for each model and add percentage difference column
|
# Add display of FP16 values for each model and add percentage difference column
|
||||||
|
|
||||||
task = 'ppl_result'
|
latest_csv['ppl_result_FP16'] = latest_csv['Model'].apply(lambda model: fp16_dict.get(model, {}).get('ppl_result', 'N/A'))
|
||||||
latest_csv[f'{task}_FP16'] = latest_csv['Model'].apply(lambda model: fp16_dict.get(model, {}).get(task, 'N/A'))
|
latest_csv['ppl_result_diff_FP16(%)'] = latest_csv.apply(lambda row: calculate_percentage_difference(row['ppl_result'], row['ppl_result_FP16']), axis=1)
|
||||||
latest_csv[f'{task}_diff_FP16(%)'] = latest_csv.apply(lambda row: calculate_percentage_difference(row[task], row[f'{task}_FP16']), axis=1)
|
|
||||||
print(csv_files)
|
print(csv_files)
|
||||||
|
|
||||||
if len(csv_files)>1:
|
if len(csv_files)>1:
|
||||||
|
|
@ -169,13 +168,8 @@ def main():
|
||||||
|
|
||||||
diffs_within_normal_range = is_diffs_within_normal_range(diff_ppl_result, threshold=highlight_threshold)
|
diffs_within_normal_range = is_diffs_within_normal_range(diff_ppl_result, threshold=highlight_threshold)
|
||||||
|
|
||||||
subset1='ppl_result_diff_last(%)'
|
|
||||||
|
|
||||||
# columns will be different: columns_sec={'ppl_result': '{:.2f}'}
|
|
||||||
columns={'ppl_result': '{:.2f}', 'last_ppl_result': '{:.2f}', 'ppl_result_diff_last(%)': '{:.2f}'}
|
columns={'ppl_result': '{:.2f}', 'last_ppl_result': '{:.2f}', 'ppl_result_diff_last(%)': '{:.2f}'}
|
||||||
# This is the same
|
|
||||||
latest_csv.drop('Index', axis=1, inplace=True)
|
latest_csv.drop('Index', axis=1, inplace=True)
|
||||||
# subset is different
|
|
||||||
|
|
||||||
styled_df = latest_csv.style.format(columns).applymap(lambda val: highlight_vals(val, max=highlight_threshold, is_last=True), subset=['ppl_result_diff_last(%)'])
|
styled_df = latest_csv.style.format(columns).applymap(lambda val: highlight_vals(val, max=highlight_threshold, is_last=True), subset=['ppl_result_diff_last(%)'])
|
||||||
styled_df = styled_df.applymap(lambda val: highlight_vals(val, max=highlight_threshold, is_last=False), subset=['ppl_result_diff_FP16(%)'])
|
styled_df = styled_df.applymap(lambda val: highlight_vals(val, max=highlight_threshold, is_last=False), subset=['ppl_result_diff_FP16(%)'])
|
||||||
|
|
@ -203,14 +197,3 @@ def main():
|
||||||
|
|
||||||
if __name__ == "__main__":
|
if __name__ == "__main__":
|
||||||
sys.exit(main())
|
sys.exit(main())
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
|
||||||
Loading…
Reference in a new issue