Skip to content

Instantly share code, notes, and snippets.

@NMZivkovic

NMZivkovic/create_model.py

Last active February 10, 2018 15:29
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 NMZivkovic/be9b305c66e53d136f89c603e414fb4a to your computer and use it in GitHub Desktop.
Save NMZivkovic/be9b305c66e53d136f89c603e414fb4a to your computer and use it in GitHub Desktop.
Download ZIP
Raw
create_model.py
# Creating a model
model = Sequential()
model.add(Dense(10, input_dim=4, activation='relu'))
model.add(Dense(10, activation='relu'))
model.add(Dense(3, activation='softmax'))
# Compiling model
model.compile(loss='categorical_crossentropy', optimizer='adam', metrics=['accuracy'])
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment