From 2ab14fa96a6c5484b5e4ba8337551bb8dcc79cc5 Mon Sep 17 00:00:00 2001 From: Christian Krinitsin Date: Tue, 3 Jun 2025 12:03:38 +0000 Subject: measure runtime of classifier --- classification/main.py | 5 +++++ 1 file changed, 5 insertions(+) (limited to 'classification/main.py') diff --git a/classification/main.py b/classification/main.py index ea4724759..46ce6488c 100755 --- a/classification/main.py +++ b/classification/main.py @@ -1,5 +1,7 @@ from transformers import pipeline from os import path, listdir, makedirs +from datetime import timedelta +from time import monotonic from argparse import ArgumentParser parser = ArgumentParser(prog='main.py') @@ -58,4 +60,7 @@ def main(): output(text, category, result['labels'], result['scores'], path.basename(bug)) if __name__ == "__main__": + start_time = monotonic() main() + end_time = monotonic() + print(timedelta(seconds=end_time - start_time)) -- cgit 1.4.1