diff --git a/dev-0/out.tsv b/dev-0/out.tsv index e2d0279..d305b39 100644 --- a/dev-0/out.tsv +++ b/dev-0/out.tsv @@ -780,7 +780,7 @@ 0 0 0 -1 +0 0 0 0 @@ -1957,7 +1957,6 @@ 0 0 0 -1 0 0 0 @@ -2073,7 +2072,6 @@ 0 0 0 -1 0 0 0 @@ -2108,288 +2106,6 @@ 0 0 0 -1 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -1 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -1 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -1 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 -0 0 0 0 @@ -2665,7 +2381,291 @@ 0 0 0 -1 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 +0 0 0 0 @@ -3519,7 +3519,7 @@ 0 0 0 -1 +0 0 0 0 @@ -4265,7 +4265,7 @@ 0 0 0 -1 +0 0 0 0 diff --git a/model.py b/model.py index f879488..09ae695 100644 --- a/model.py +++ b/model.py @@ -16,7 +16,10 @@ class Model(nn.Module): self.output_dim = output_dim self.fc1 = nn.Linear(self.input_dim, self.hidden_dim) - self.fc2 = nn.Linear(self.hidden_dim, self.output_dim) + self.fc2 = nn.Linear(self.hidden_dim, self.hidden_dim) + self.fc3 = nn.Linear(self.hidden_dim, self.output_dim) + + self.relu = nn.ReLU() self.criterion = nn.BCELoss() self.optimizer = torch.optim.SGD(self.parameters(), lr=0.01) @@ -25,14 +28,19 @@ class Model(nn.Module): """Step forward learning fn""" x = self.fc1(x) - x = torch.relu(x) + x = self.relu(x) x = self.fc2(x) + x = self.relu(x) + x = self.fc3(x) x = torch.sigmoid(x) return x def run_training(self, X_train, Y_train, batch_size, epochs_count): for _ in range(epochs_count): self.train() + print(f"{Y_train.shape[0]}, {Y_train.shape[0] == self.input_dim}") + print(f"{Y_train.shape[0]}, {Y_train.shape[0] == self.hidden_dim}") + print(f"{Y_train.shape[0]}, {Y_train.shape[0] == self.output_dim}") for i in range(0, Y_train.shape[0], batch_size): X = X_train[i: i + batch_size] X = torch.tensor(X)