Skip to content

Instantly share code, notes, and snippets.

Show Gist options
  • Save javaeeeee/71ed9bedea08f274f494fb18a744b0f5 to your computer and use it in GitHub Desktop.
Save javaeeeee/71ed9bedea08f274f494fb18a744b0f5 to your computer and use it in GitHub Desktop.
Tensorflow Binary Classification Model
model = Sequential()
model.add(normalizer)
model.add(Dense(5, input_shape=(X_train.shape[1],),
activation='relu', kernel_initializer='he_normal'))
model.add(Dropout(0.2))
model.add(Dense(5, activation="relu",
kernel_initializer='he_normal'))
model.add(Dropout(0.2))
model.add(Dense(1, activation="sigmoid",
kernel_initializer="glorot_normal"))
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment