36 lines
1.4 KiB
Python
36 lines
1.4 KiB
Python
m = ['żona', 'żony', 'żonie', 'żoną', 'zona', 'zony', 'zonie', 'mecz', 'komputer', 'serwer',
|
|
'ziomek', 'ziom', 'ziomków', 'ziomkow', 'kumpel', 'kolega', 'kolegą', 'kolegi'
|
|
'lagi', 'gram', 'gra', 'gry', 'diablo', 'wowa', 'tibia', 'tibie', 'samochód', 'samochod', 'auto', 'felga', 'felgi',
|
|
'bmw', 'audi', 'passat', 'informatyka', 'informatykę', 'polibuda', 'polibudzie', 'politechnika', 'politechnice',
|
|
'piwo', 'laska', 'laseczka', 'laseczki', 'dziewczynie', 'siema', 'elo', 'yo', 'spadam', 'windows',
|
|
'linux', 'sterowniki', 'silka', 'siłka', 'wyciskam', 'klate', 'paliwo', 'paliwa', 'lv', 'eq',
|
|
]
|
|
|
|
|
|
f = ['mąż', 'okres', 'ciąża', 'kocham', 'chłopak', 'torebka', 'torebke', 'podpaska', 'podpaski',
|
|
'ciąża', 'ciąży', 'ciaza', 'ciazy', 'sukienka',
|
|
]
|
|
|
|
|
|
def calculate(p_in, p_out):
|
|
results = []
|
|
with open(p_in, encoding='utf-8') as file:
|
|
for line in file.readlines():
|
|
words = line.split()
|
|
m_score = len([w for w in words if w in m])
|
|
f_score = len([w for w in words if w in f])
|
|
|
|
if m_score > f_score:
|
|
results.append('1')
|
|
else:
|
|
results.append('0')
|
|
|
|
with open(p_out, 'w') as file:
|
|
for r in results:
|
|
file.write(r + '\n')
|
|
|
|
|
|
calculate('dev-0/in.tsv', 'dev-0/out.tsv')
|
|
calculate('dev-1/in.tsv', 'dev-1/out.tsv')
|
|
calculate('test-A/in.tsv', 'test-A/out.tsv')
|