s152483/paranormal-or-skeptic
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Merge branch 'ISI-33' of git.wmi.amu.edu.pl:s152483/paranormal-or-skeptic into ISI-33

Browse Source
This commit is contained in:
Łukasz Szymula 2020-04-27 16:26:01 +02:00
commit 634a2cb69a
3 changed files with 10448 additions and 10456 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10545
dev-0/out.tsv
View File

File diff suppressed because it is too large Load Diff

24
predict_rm.py
View File

@ -3,28 +3,22 @@
import pandas as pd
import csv
import pickle
import numpy as np
def predict():
dev0 = pd.read_csv("dev-0/in.tsv", delimiter="\t", header=None, names=["document","date"], quoting=csv.QUOTE_NONE)
testA = pd.read_csv("test-A/in.tsv", delimiter="\t", header=None, names=["document","date"], quoting=csv.QUOTE_NONE)
devdoc = dev["document"]
testdoc = testA["document"]
dev0 = pd.read_csv("dev-0/in.tsv", delimiter="\t", header=None, names=["document","date"], quoting=csv.QUOTE_NONE)["document"]
testA = pd.read_csv("test-A/in.tsv", delimiter="\t", header=None, names=["document","date"], quoting=csv.QUOTE_NONE)["document"]
clf = pickle.load(open("clf.model", "rb"))
vectorizer = pickle.load(open("vectorizer.model", "rb"))
dev0_vectorizer = vectorizer.transform(devdoc)
testA_vectorizer = vectorizer.transform(testdoc)
dev0_vectorizer = vectorizer.transform(dev0)
testA_vectorizer = vectorizer.transform(testA)
y_dev = clf.predict_proba(dev0_vectorizer)
y_test = clf.predict_proba(testA_vectorizer)
y_dev = clf.predict_proba(dev0_vectorizer)[:, 1]
y_test = clf.predict_proba(testA_vectorizer)[:, 1]
with open("dev-0/out.tsv", "w") as devout:
for line in y_dev:
devout.write(str(line[1])+"\n")
with open("test-A/out.tsv", "w") as testaout:
for line in y_test:
testaout.write(str(line[1])+"\n")
np.savetxt('test-A/out.tsv', y_test, '%f')
np.savetxt('dev-0/out.tsv', y_dev, '%f')
predict()

10305
test-A/out.tsv
View File

File diff suppressed because it is too large Load Diff