s444415/sport-text-classification
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

s444415

Browse Source
This commit is contained in:
mkozlowskiAzimuthe 2022-05-18 16:38:52 +02:00
parent d3d2656597
commit ab3072e91a
1 changed files with 0 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
run.py
View File

@ -3,7 +3,6 @@ from sklearn.naive_bayes import GaussianNB
from sklearn.feature_extraction.text import TfidfVectorizer from sklearn.feature_extraction.text import TfidfVectorizer
def read_train_data(): def read_train_data():
print('Load train data') print('Load train data')
train_data = pd.read_csv("train/train.tsv", sep='\t', header=None) train_data = pd.read_csv("train/train.tsv", sep='\t', header=None)
@ -38,5 +37,4 @@ def get_result():
x, x_p = vectorize(x, x_p) x, x_p = vectorize(x, x_p)
return calc_score(x, y, x_p) return calc_score(x, y, x_p)
pd.DataFrame(get_result()).to_csv("dev-0/out.tsv", header=False, index=None) pd.DataFrame(get_result()).to_csv("dev-0/out.tsv", header=False, index=None)