2f00e2570460fe0a6acd8db94c4d8624b3ddd1eb,inferno/tests/test_net.py,MyModule,forward,#MyModule#Any#,33

Before Change

    def forward(self, X, **kwargs):
        X = self.nonlin(self.dense0(X))
        X = self.dropout(X)
        X = F.relu(self.dense1(X))
        X = F.softmax(self.output(X))
        return X


class TestNeuralNet:

After Change

    def forward(self, X, **kwargs):
        X = self.nonlin(self.dense0(X))
        X = self.dropout(X)
        X = self.dense1(X)
        X = self.output(X)
        return X


class TestNeuralNetRegressor:

Instances