From af73e7ad05181c4e9187bac51010cf7e51d0ab34 Mon Sep 17 00:00:00 2001 From: Jan Przybylski Date: Mon, 10 May 2021 23:25:39 +0200 Subject: [PATCH] up --- program.py | 11 ++++++----- 1 file changed, 6 insertions(+), 5 deletions(-) diff --git a/program.py b/program.py index c61dc3b..6335a1a 100644 --- a/program.py +++ b/program.py @@ -5,7 +5,7 @@ from sklearn import preprocessing import pandas as pd from gensim.test.utils import common_texts from gensim.models import Word2Vec - +00 with open("train/in.tsv") as f: X_train = f.readlines() with open("train/expected.tsv") as ff: @@ -13,10 +13,11 @@ with open("train/in.tsv") as f: with open("test-A/in.tsv") as d: X = d.readlines() -model = Word2Vec(X_train, min_count=1,size= 50,workers=3, window =3, sg = 1) -X_train=model[X_train] -X=model[X] -FEAUTERES=1000 +model = Word2Vec(X_train, min_count=1,size= 500,workers=3, window =3, sg = 1) +model1 = Word2Vec(X, min_count=1,size= 500,workers=3, window =3, sg = 1) +X_train=model.wv +X=model1.wv +FEAUTERES=500 class NeuralNetworkModel(torch.nn.Module):