beijinggao / laogaoganhuo

Record all the valuable things!
1 stars 0 forks source link

tensorflow cosine similarity #3

Open beijinggao opened 7 years ago

beijinggao commented 7 years ago

https://github.com/tensorflow/tensorflow/blob/master/tensorflow/examples/tutorials/word2vec/word2vec_basic.py

beijinggao commented 7 years ago

optimizer = tf.train.GradientDescentOptimizer(1.0).minimize(loss)

 Compute the cosine similarity between minibatch examples and all embeddings.

norm = tf.sqrt(tf.reduce_sum(tf.square(embeddings), 1, keep_dims=True)) normalized_embeddings = embeddings / norm valid_embeddings = tf.nn.embedding_lookup( normalized_embeddings, valid_dataset) similarity = tf.matmul(valid_embeddings, normalized_embeddings, transpose_b=True)