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:
Please be nice still learning!