From 541577f302a996275076abe6e167973858da1ed3 Mon Sep 17 00:00:00 2001 From: Kornelia Girejko Date: Tue, 26 Apr 2022 22:10:45 +0200 Subject: [PATCH] run.py --- run.py | 35 +++++++++++++++++++++++++++++++++++ 1 file changed, 35 insertions(+) create mode 100644 run.py diff --git a/run.py b/run.py new file mode 100644 index 0000000..0407659 --- /dev/null +++ b/run.py @@ -0,0 +1,35 @@ +# słowniki +male = ['gry', 'grać', 'grami', 'auto', 'samochód', 'silnik', 'kolegami', 'brodę', 'zarost', 'samochodem', 'piwo', + 'kolega', 'fura', 'kasa', 'pieniądze', 'autem', 'mechanika', 'piłka', 'mecz', 'nożna', 'sport', 'motor', + 'skuter', 'pograć', 'alkohol', 'ubuntu', 'systemu', 'serwer', 'linux', 'windows', 'procesor', 'dysk', 'gb', + 'mb'] + +female = ['ja', 'koleżanka', 'koleżankami', 'koleżanką', 'przyjaciółki', 'przyjaciółka', 'przyjaciółkami', + 'kosmetyczka', 'okres', 'praca', 'paznokcie', 'włosy', 'usta', 'ciąża', 'lekarz', 'lekarza', 'ciąży', + 'dziecko', 'dzieci', 'wino', 'sukienka', 'sukienkę', 'torebka', 'tprebkę', 'makijaż', 'pomadka', 'piersi', + 'wesele', 'ślub'] + + +# klasyfikacja +def calculation(path_in, path_out): + results = [] + with open(path_in, encoding='utf-8') as file: + for line in file.readlines(): + line = line.replace('.', ' ').replace(',', ' ').lower() + words = line.split() + male_score = len([w for w in words if w in male]) + female_score = len([w for w in words if w in female]) + + if male_score > female_score: + results.append('1') + else: + results.append('0') + + with open(path_out, 'w') as file: + for r in results: + file.write(r + '\n') + + +calculation('dev-0/in.tsv', 'dev-0/out.tsv') +calculation('dev-1/in.tsv', 'dev-1/out.tsv') +calculation('test-A/in.tsv', 'test-A/out.tsv') \ No newline at end of file