Code example

[iprada]

A group had issues with the dimensionality of the model due to lack of clarity in the specification of the rows. I share the model here.

import torch.nn as nn
import torch.nn.functional as F


class Net(nn.Module):
    def __init__(self):
        super().__init__()
        # 2 fully connected layers 
        self.fc1 = nn.Linear(in_features=5001, out_features=1000)
        self.fc2 = nn.Linear(in_features=1000, out_features=200)
        self.fc3 = nn.Linear(in_features=200, out_features=2)

    def forward(self, x):
        # pass data through the layers
        x = F.relu(self.fc1(x))
        x = F.relu(self.fc2(x))
        x = self.fc3(x)
        return x