Commit 929ec5a4 authored by Matúš Žiak's avatar Matúš Žiak
Browse files

Added train data and model is trained 5 times

parent 05c2039b
......@@ -16,19 +16,27 @@ def stemmed_phrase(text):
ps = nltk.stem.PorterStemmer()
return ' '.join([ps.stem(w) for w in filtered_sent])
clf_list = []
data = pd.read_csv('train.tsv', sep='\t')
X_train = [stemmed_phrase(phrase) for phrase in data['Phrase']]
for i in range(1, 6):
data = pd.read_csv('train' + str(i) + '.tsv', sep='\t')
X_train = [stemmed_phrase(phrase) for phrase in data['Phrase']]
tf = TfidfVectorizer()
text_tf = tf.fit_transform(X_train)
clf = MultinomialNB().fit(text_tf, data['Sentiment'])
tf = TfidfVectorizer()
text_tf = tf.fit_transform(X_train)
clf = MultinomialNB().fit(text_tf, data['Sentiment'])
clf_list.append(clf)
def predict(phrase):
stemmed = stemmed_phrase(phrase)
transformed = tf.transform([stemmed])
return clf.predict(transformed)[0]
result = 0
for clf in clf_list:
result += clf.predict(transformed)[0]
return result
SQL_SELECT = 'SELECT TOP 100 * FROM [data] WHERE bank_robbery IS NULL ORDER BY timestamp'
......
Sentiment Phrase
1 Prepadnutie banky.
1 Nevzdorujte.
1 O nic sa nepokusajte
1 Doneste mi vsetky peniaze
1 Vsetci budete ticho!
1 Budem strielat!
1 Zabijem ta ked nieco skusis.
1 Peniaze do tasky.
0 Pred tyzdnom som bol v banke.
0 Dajme vsetky peniaze na ucet.
0 Chceli by sme si otvorit spolocny ucet.
0 Hypotekarny uver je velmi vyhodny.
0 Co potrebujem na vybavenie pozicky.
0 Len obciansky preukaz.
0 Prihlasite sa vasim hlasom.
0 Stiahnite si aplikaciu tatra banky.
0 Kolegyna vam poradi.
0 Pridte zajtra.
0 Nemusite sa o to vobec bat, my za vas vsetko vybavime.
0 Ako vam pomozem?
Sentiment Phrase
1 Ruky hore!
1 Vsetci lahnut na zem.
1 Kazdy si teraz pomaly lahne.
1 Tvarou k zemi.
1 Ruky drzte tak aby som ich videl.
1 Kazdy ticho a nikomu sa nic nestane.
1 Ticho lebo budem strielat.
1 Pokuste sa o nieco a ja vas zastrelim.
1 Nic neskusaj lebo ta zabijem.
0 U nas to mate bezplatne.
0 Nie, nie to je v poriadku.
0 Ako to funguje, kedy neplatim nic?
0 Co poviete na studentsky ucet bez poplatkov?
0 Mate nanho narok do konca studia.
0 Poprosim vas podpis sem a sem.
0 A este jeden.
0 Kolko by ste chceli vlozit?
0 Chcela by som vlozit sedem tisic.
0 Vas urok bude dve percenta mesacne.
Sentiment Phrase
1 Toto je prepad
1 Toto je prepad banky, vsetci ruky hore!
1 Vsetci dajte mobily do tejto tasky a lahnite si!
1 Dajte mi vsetky peniaze z trezoru!
1 Vsetci ostante ticho!
1 Nevolajte policiu, inak zastrelim niekoho!
1 Pomaly si vsetci lahnite.
1 Tasku s peniazmi polozte na stol.
0 Vedel si o tom, ze minule vylupili banku v Kosiciach?
0 Potrebujem neskor aj vybrat peniaze.
0 Minule v banke som stala v rade hodinu.
0 Si predstav co by sa stalo, keby teraz prisli vykradnut banku.
0 Musim ist kupit este rozky a maslo.
0 Som v banke, nemozem teraz hovorit.
0 Chcela by som vlozit peniaze.
0 Nasla som v bankomate 100 eur.
0 Nevykrikuj, v banke musis byt ticho.
0 Nech sa paci, dalsi.
0 Kolko by ste chceli vybrat?
0 Prosim zavolaj mi neskor, akurat ma volaju k priehradke v banke.
0 Chcela by som zmenit 50 eur na forinty.
0 Aky je teraz kurz na ceske koruny?
Sentiment Phrase
1 Prepad!
1 Hore ruky!
1 Ruky hore!
1 Otvor trezor.
1 Na zem!
1 Mobily sem!
1 Peniaze do tasky!
1 Naval prachy!
1 Chcem prachy!
1 Sem prachy!
1 Chcem peniaze!
1 Ticho!
1 Ani slovo!
1 Ziadna policia!
1 Toto je prepad!
1 Vsetci lahnut na zem!
1 Daj do tej tasky peniaze.
1 Tu daj prachy.
1 Vsetci ticho!
1 Nevolat nikoho, lebo budem strielat.
1 Niekto si to odnesie.
1 Neopovaz sa niekoho volat.
0 Traja ho prepadli zo zadu.
0 Poprosim vas peniaze dajte sem.
0 Banka je otvorena od rana.
0 Dobry den.
0 Co si prajete?
0 Vcera vykradli banku.
0 Spadlo mi pero na zem.
0 Som v banke, nemozem teraz hovorit.
0 Nepocujem vas.
0 Nevykrikuj, v banke musis byt ticho.
0 Nech sa paci, dalsi.
0 Kolko by ste chceli vybrat?
0 Prosim zavolaj mi neskor, akurat som na rade.
0 Chcela by som zajst do potravin.
0 Uvidim aka bude odozva.
0 Takto by sa z nich stali efektivnejsi clenovia spolocnosti na slovensku.
0 Uz su v penazenke.
0 Doma to urobim.
0 Co by si bezo mna robila?
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment