From b372f5a5601734a150a32651c507a1e9ead575a5 Mon Sep 17 00:00:00 2001 From: Christian Krinitsin Date: Sun, 1 Jun 2025 14:54:03 +0200 Subject: classifier: iterates through mailing list --- classification/test.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) mode change 100644 => 100755 classification/test.py (limited to 'classification/test.py') diff --git a/classification/test.py b/classification/test.py old mode 100644 new mode 100755 index a8bc7e1ee..bcd6b4392 --- a/classification/test.py +++ b/classification/test.py @@ -2,16 +2,15 @@ from os import listdir, path directory : str = "./test_input" -def test(classifier): +def test(classifier, categories): for name in listdir(directory): if name == "README.md": continue with open(path.join(directory, name), "r") as file: - sequence_to_classify = file.read() + text = file.read() - candidate_labels = ['semantic', 'other', 'mistranslation', 'instruction'] - result = classifier(sequence_to_classify, candidate_labels, multi_label=True) + result = classifier(text, categories, multi_label=True) print(name) for label, score in zip(result["labels"], result["scores"]): -- cgit 1.4.1