petite-difference-challenge2/dev-0/model.ipynb
Adrian Charkiewicz e310d99b89 51%
2022-04-13 16:58:26 +02:00

5.1 KiB
Raw Blame History

import pandas as pd
import csv
tsv_data = pd.read_csv('in.tsv', sep='\t',header=None, quoting=csv.QUOTE_NONE)[0]
expected = pd.read_csv('expected.tsv', sep='\t',header=None)[0]
print(len(expected))
print(len(tsv_data))
137314
137314
male={'silnik', 'windows', 'gb', 'mb', 'mecz', 'pc', 'opony', 'apple', 'iphone', 'zwiastuny', 'hd', 'ubuntu', 'system', 'serwer'}
female={'ciąża', 'miesiączki', 'ciasto', 'ciąże', 'zadowolona', 'antykoncepcyjne', 'ginekologia', 'tabletki', 'porodzie', 'mąż', 'krwawienie', 'ciasta'}
male = {x[:6].lower() for x in male}
female = {x[:6].lower() for x in female}
trimmed_docs=[]
for document in tsv_data:
    new_doc=[]
    for word in str(document).lower().split():
        new_doc.append(word[:6])
    trimmed_docs.append(new_doc)
male_or_female=[]
for doc in trimmed_docs:
    male_or_female.append((len(male&set(doc)), len(female&set(doc))))
answers=[]
for i in male_or_female:
    if i[0]>i[1]:
        answers.append(1)
    else:
        answers.append(0)
result=[]
for i in range(len(answers)):
    if answers[i]==expected[i]:
        result.append(1)
    else:
        result.append(0)
print(f'Predykcja modelu wynosi {sum(result)/len(result)*100:.6f}%')
Predykcja modelu wynosi 51.007909%
trimmed_docs[0]
['cierpi',
 'na',
 'strasz',
 'lagi',
 '',
 'kilkan',
 'sekund',
 'lub',
 'dłużej',
 'czarne',
 'ekranu',
 'przy',
 'próbie',
 'przełą',
 'się',
 '/',
 'urucho',
 'prawie',
 'każdej',
 'aplika',
 'dodatk',
 'telefo',
 'mi',
 'się',
 'wyłącz',
 'czasem',
 'bez',
 'powodu',
 '',
 'sam',
 'z',
 'siebie',
 'albo',
 'reseto',
 'ostatn',
 'nawet',
 'przegl',
 'zaczęł',
 'się',
 'często',
 'zawies',
 'i',
 'androi',
 'propon',
 'wymusz',
 'zamkni',
 'do',
 'tego',
 'te',
 'proble',
 'z',
 'połącz',
 'do',
 'komput',
 'przez',
 'usb.']