kubapok/paranormal-or-skeptic-ISI-public
1
0
Fork 51
Code Issues Pull Requests Releases Wiki Activity

second attempt

Browse Source
This commit is contained in:
pietrzakkuba 2021-05-24 21:01:20 +02:00
parent 76fffb4848
commit 3e0857c6c7
3 changed files with 526 additions and 526 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

594
dev-0/out.tsv
View File

File diff suppressed because it is too large Load Diff

6
log-reg.py
View File

@ -55,9 +55,9 @@ test_x = [get_word2vec(document) for document in test_x]
print('model training') print('model training')
network = MyNeuralNetwork(300, 600, 1) network = MyNeuralNetwork(300, 600, 1)
criterion = torch.nn.BCELoss() criterion = torch.nn.BCELoss()
optimizer = torch.optim.SGD(network.parameters(), lr=0.1) optimizer = torch.optim.SGD(network.parameters(), lr=0.02)
epochs = 5 epochs = 15
batch_size = 5 batch_size = 15
for epoch in range(epochs): for epoch in range(epochs):
network.train() network.train()

452
test-A/out.tsv
View File

File diff suppressed because it is too large Load Diff