ayo/ipex-llm
1
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 1

Mark Color Modification

Browse source
This commit is contained in:
jenniew 2024-04-12 14:00:50 +08:00
parent 9bbfcaf736
commit cdbb1de972
2 changed files with 5 additions and 22 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
python/llm/dev/benchmark/perplexity/make_table.py
View file

@ -35,8 +35,8 @@ def make_table(result_dict):
"""Generate table of results."""
md_writer = MarkdownTableWriter()
latex_writer = LatexTableWriter()
md_writer.headers = ["Model", "Precision", "ppl_result"]#["Model", "Precision", "en", "zh"]
latex_writer.headers = ["Model", "Precision", "ppl_result"]#"en", "zh"]
md_writer.headers = ["Model", "Precision", "ppl_result"]
latex_writer.headers = ["Model", "Precision", "ppl_result"]
languages = ["en", "zh"]
values = []
@ -46,7 +46,7 @@ def make_table(result_dict):
for language in languages:
task_results = prec_results.get(language, None)
if task_results is None:
continue#value.append("")
continue
else:
result = task_results["results"]
value.append("%.4f" % result)

21
python/llm/test/benchmark/perplexity/ppl_csv_to_html.py
View file

@ -106,9 +106,8 @@ def main():
# Add display of FP16 values for each model and add percentage difference column
task = 'ppl_result'
latest_csv[f'{task}_FP16'] = latest_csv['Model'].apply(lambda model: fp16_dict.get(model, {}).get(task, 'N/A'))
latest_csv[f'{task}_diff_FP16(%)'] = latest_csv.apply(lambda row: calculate_percentage_difference(row[task], row[f'{task}_FP16']), axis=1)
latest_csv['ppl_result_FP16'] = latest_csv['Model'].apply(lambda model: fp16_dict.get(model, {}).get('ppl_result', '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)
print(csv_files)
if len(csv_files)>1:
@ -168,14 +167,9 @@ def main():
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}'}
# This is the same
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 = 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__":
sys.exit(main())