INNER CODE UNIT · Python

forward

maxpumperla/learning_ray · notebooks/fare_predictor.py:19

    def forward(self, x):
        x = F.relu(self.fc1(x))
        x = self.bn1(x)
        x = F.relu(self.fc2(x))
        x = self.bn2(x)
        x = torch.sigmoid(self.fc3(x))

        return x

View source record →