A little change

This commit is contained in:
Blade He 2024-10-03 16:31:16 -05:00
parent f0dd7f9e89
commit b18c48efeb
1 changed files with 21 additions and 5 deletions

26
main.py
View File

@ -446,7 +446,7 @@ def get_metrics(
ground_truth_sheet_name=ground_truth_sheet_name, ground_truth_sheet_name=ground_truth_sheet_name,
output_folder=output_folder, output_folder=output_folder,
) )
missing_error_list, metrics_list, metrics_file = metrics.get_metrics(strict_model=False) missing_error_list, metrics_list, metrics_file = metrics.get_metrics(strict_model=True)
return missing_error_list, metrics_list, metrics_file return missing_error_list, metrics_list, metrics_file
@ -681,7 +681,6 @@ if __name__ == "__main__":
"333207452", "333207452",
"334718372", "334718372",
"344636875", "344636875",
# "349679479",
"362246081", "362246081",
"366179419", "366179419",
"380945052", "380945052",
@ -701,15 +700,32 @@ if __name__ == "__main__":
"448265376", "448265376",
"449555622", "449555622",
"449623976", "449623976",
"469138353" "458291624",
"458359181",
"463081566",
"469138353",
"471641628",
"476492237",
"478585901",
"478586066",
"479042264",
"479042269",
"479793787",
"481475385",
"483617247",
"486378555",
"486383912",
"492121213",
"497497599",
"502693599"
] ]
# special_doc_id_list = check_mapping_doc_id_list # special_doc_id_list = check_mapping_doc_id_list
special_doc_id_list = check_db_mapping_doc_id_list special_doc_id_list = check_db_mapping_doc_id_list
# special_doc_id_list = ["380945052", "382366116", "387202452", "394778487", "469138353"] # special_doc_id_list = ["503194284"]
output_mapping_child_folder = r"/data/emea_ar/output/mapping_data/docs/" output_mapping_child_folder = r"/data/emea_ar/output/mapping_data/docs/"
output_mapping_total_folder = r"/data/emea_ar/output/mapping_data/total/" output_mapping_total_folder = r"/data/emea_ar/output/mapping_data/total/"
re_run_extract_data = False re_run_extract_data = False
re_run_mapping_data = True re_run_mapping_data = False
force_save_total_data = True force_save_total_data = True
extract_ways = ["text"] extract_ways = ["text"]