run.py

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')