Skip to content

Instantly share code, notes, and snippets.

@oscar-defelice
Created November 4, 2020 08:28
Show Gist options
  • Save oscar-defelice/3914cfee97997b2f93201adccc433b26 to your computer and use it in GitHub Desktop.
Save oscar-defelice/3914cfee97997b2f93201adccc433b26 to your computer and use it in GitHub Desktop.
emb_dim = 64
dropout_rate = 0.3
n_labels = y.shape[1]
learning_rate = 0.0006
loss = tf.keras.losses.CategoricalCrossentropy(from_logits=True)
metric = tf.keras.metrics.CategoricalAccuracy('accuracy')
opt = tf.keras.optimizers.Adam(learning_rate = learning_rate)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment