s426228/paranormal-or-skeptic-ISI-public-2
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

update

Browse Source
This commit is contained in:
Jan Przybylski 2021-05-10 23:07:13 +02:00
parent 09b477c5a4
commit 670ac37c3f
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
program.py
View File

@ -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)