Skip to content

Instantly share code, notes, and snippets.

@shayaf84
Last active September 30, 2021 21:45
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save shayaf84/15127c8ff428533ae58efd0bc6fdcc18 to your computer and use it in GitHub Desktop.
Save shayaf84/15127c8ff428533ae58efd0bc6fdcc18 to your computer and use it in GitHub Desktop.
from tensorflow.keras.models import Sequential
from tensorflow.keras.layers import add, Dense, Dropout
from tensorflow.keras.optimizers import Adam
model = Sequential()
model.add(Dense(64,input_dim=29,activation='relu'))
model.add(Dense(32, activation = 'relu'))
model.add(Dropout(0.5))
model.add(Dense(16,activation = 'relu'))
model.add(Dropout(0.15))
model.add(Dense(1, activation = 'sigmoid'))
opt = Adam(learning_rate=1e-4, decay=1e-6)
model.compile(loss="binary_crossentropy",optimizer=opt,metrics=['Precision','Recall'])
model.fit(X_train, y_train, epochs = 5,\
validation_data = (X_test, y_test))
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment