Skip to content

Instantly share code, notes, and snippets.

@this-is-richard

this-is-richard/single_label_categorical_classification.py

Created April 25, 2019 05:36
Show Gist options

  • Select an option

    Learn more about clone URLs
  • Save this-is-richard/3d94d06c4000b5187ecce69ad7fcd4d2 to your computer and use it in GitHub Desktop.

Select an option

Learn more about clone URLs
Save this-is-richard/3d94d06c4000b5187ecce69ad7fcd4d2 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
single_label_categorical_classification.py
model = models.Sequential()
model.add(layers.Dense(32, activation='relu', input_shape=(num_input_features,)))
model.add(layers.Dense(32, activation='relu'))
model.add(layers.Dense(num_classes, activation='softmax'))
model.compile(optimizer='rmsprop', loss='categorical_crossentropy')
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment