Skip to content

Instantly share code, notes, and snippets.

@hadifar

hadifar/tf_imdb_embed.py

Created December 6, 2018 08:49
Show Gist options
  • Save hadifar/b9b9b771a36f644cfaf530eb0d6a7451 to your computer and use it in GitHub Desktop.
Save hadifar/b9b9b771a36f644cfaf530eb0d6a7451 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
tf_imdb_embed.py
embeddings_matrix = tf.get_variable("embedding", [vocabulary_size, embedding_size])
embed = tf.nn.embedding_lookup(embeddings_matrix, x)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment