From 670ac37c3fc10d5d567874a4f93e506c943aa37a Mon Sep 17 00:00:00 2001 From: Jan Przybylski Date: Mon, 10 May 2021 23:07:13 +0200 Subject: [PATCH] update --- program.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/program.py b/program.py index 2dd42fc..56f743e 100644 --- a/program.py +++ b/program.py @@ -16,7 +16,7 @@ with open("train/in.tsv") as f: model = Word2Vec(X_train, min_count=1,size= 50,workers=3, window =3, sg = 1) X_train=model[X_train] X=model[X] -FEAUTERES=10000 +FEAUTERES=1000 class NeuralNetworkModel(torch.nn.Module): def __init__(self): @@ -60,7 +60,7 @@ for epoch in range(10): loss_score += loss.item() * Y.shape[0] - with open('test-A/out1.tsv', 'w') as file: + with open('test-A/out.tsv', 'w') as file: for e in Y_predictions: file.write("%f\n" % e)