From bf49cfef2f923bcaaf6e4f44470b0c29cb829c34 Mon Sep 17 00:00:00 2001 From: Christian Krinitsin Date: Mon, 30 Jun 2025 15:22:40 +0200 Subject: add analyze_results script and results for 118 and deepseek-2-tmp --- classification/tools/analyze_results.py | 48 +++++++++++++++++++++++++++++++++ 1 file changed, 48 insertions(+) create mode 100755 classification/tools/analyze_results.py (limited to 'classification') diff --git a/classification/tools/analyze_results.py b/classification/tools/analyze_results.py new file mode 100755 index 000000000..663c1f559 --- /dev/null +++ b/classification/tools/analyze_results.py @@ -0,0 +1,48 @@ +from argparse import ArgumentParser +from os import path, listdir + +parser = ArgumentParser() + +parser.add_argument('-b', '--bugs', required = True) +parser.add_argument('-d', '--search_directory', required = True) + +args = parser.parse_args() + +def list_files_recursive(directory, basename = False): + result = [] + for entry in listdir(directory): + full_path = path.join(directory, entry) + if path.isdir(full_path): + result = result + list_files_recursive(full_path, basename) + else: + if basename: + result.append(path.basename(full_path)) + else: + result.append(full_path) + return result + +def output_csv(dictionary, full_path): + with open(path.join(full_path, 'analysis.csv'), "w") as file: + file.write("category, count\n") + for key, value in dictionary.items(): + file.write(f"{key}, {value}\n") + +def main(): + result = {} + mistranslation_bugs = list_files_recursive(args.bugs, True) + bugs = list_files_recursive(args.search_directory, False) + + for mistranslation_bug in mistranslation_bugs: + for bug in bugs: + if mistranslation_bug == path.basename(bug): + category = path.basename(path.dirname(bug)) + if category in result: + result[category] += 1 + else: + result[category] = 1 + continue + + output_csv(result, args.search_directory) + +if __name__ == "__main__": + main() -- cgit 1.4.1