Skip to content

Instantly share code, notes, and snippets.

@Sanlap1997
Created September 21, 2020 16:18
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 Sanlap1997/575ce74af25ff0874a4140ca39d7635d to your computer and use it in GitHub Desktop.
Save Sanlap1997/575ce74af25ff0874a4140ca39d7635d to your computer and use it in GitHub Desktop.
model = Sequential()
model.add(Dense(30, activation='relu', input_shape=(train_x.shape[1],)))
model.add(Dense(train_y.shape[1], activation='softmax'))
model.compile(optimizer=tf.keras.optimizers.Adam(learning_rate=0.001), loss='categorical_crossentropy', metrics=['accuracy'])
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment