I am trying to learn PyTorch just using a basic sequential model and comparing against keras - I am getting a widely different accuracy on the titanic dataset.
“”“model_2 = torch.nn.Sequential(torch.nn.Linear(7, 16),
torch.nn.ReLU(),
torch.nn.Linear(16, 16),
torch.nn.ReLU(),
torch.nn.Linear(16, 1),
torch.nn.Sigmoid())”""
Keras:
“”“model = Sequential()
model.add(Dense(16, activation=‘relu’, input_shape=(7,)))
model.add(Dense(16, activation=‘relu’))
model.add(Dense(1, activation=‘sigmoid’))”""
colab notebook:
  
      

      colab.research.google.com
  

  
    

Google Colaboratory 3



  

  
    
    
  

  


Please be nice still learning!