from torch import nn

class BinaryClassificationWithLogits(nn.Module):
  def __init__(self,in_features,out_features, hidden_features):
    super().__init__()
    self.linear_block1=nn.Linear(in_features=in_features,out_features=hidden_features)
    self.linear_block2=nn.Linear(in_features=hidden_features,out_features=out_features)

  def forward(self,X):
  
    x=self.linear_block1(X)
   
    x=self.linear_block2(x)
   
    # return LOGITS
    return x