def forward(self, features):
    hidden = self.in_linear(features)
    hidden = self.drop(hidden)
    hidden = self.act_fn(hidden)
    predicted = self.out_linear(hidden)
    return predicted
def forward(self, features):
    hidden = self.in_linear(features)
    hidden = self.drop(hidden)
    hidden = self.act_fn(hidden)
    predicted = self.out_linear(hidden)
    return predicted