Добавить слова в качестве метки для встраивания слов - PullRequest
1 голос
/ 10 марта 2019

Я пытаюсь визуализировать мое вложение слов с помощью тензорной доски, но не могу решить проблему с меткой. Проблема в том, что у меня есть переменная, как это:

word_embeddings = tf.Variable(tf.random_uniform([vocabulary_size,word_embedding_size],-1.0,1.0))

и когда я сохраняю его для следующего использования в тензорной доске:

saver = tf.train.Saver({"word_embeddings": word_embeddings})
embeddings_save_path = os.path.join(os.getcwd(), folder_name, 'name.ckpt')
save_embeddings = saver.save(sess, embeddings_save_path)
print('Model saved')

в качестве метки я вижу только индекс (1 ... vocabulary_size)

Когда я печатаю переменную word_embeddings, я вижу только цифры, например:

[0.000 0.000 ...]
[0.000 0.000 ...]
[0.000 0.000 ...]

Как добавить метку, т.е. blue: (0.01359, 0.00075997, 0.24608, ..., -0.2524, 1.0048, 0.06259) (https://www.tensorflow.org/guide/embedding)

Я не очень знаком с этой областью, поэтому, если я не добавлю сюда всю необходимую информацию, дайте мне знать.

Я ищу здесь, но отвечаю как это не очень помогло

...