Тестирование LSTM в Tensorflow - PullRequest
Новая
       16

Тестирование LSTM в Tensorflow

0 голосов
/ 28 февраля 2019

Моя сеть сложена LSTM->FC Layer.Если мне нужно передать входные данные для тестирования, могу ли я просто передать его последнему слою FC, чтобы график подключился, извлекая соответствующие входные данные для каждого слоя, или мне нужно перестроить график?

Я сконструировал свой код примерно так: 

with tf.Session() as sess:    

    saver = tf.train.import_meta_graph(r'my_model.meta')
    saver.restore(sess,tf.train.latest_checkpoint('./'))
    graph = tf.get_default_graph()
    w1 = graph.get_tensor_by_name("batch_x:0") 
    w2 = graph.get_tensor_by_name("batch_y:0") 
    operation = graph.get_tensor_by_name("fully_connected/Relu:0")
    output=sess.run(operation,{w1:input})

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

...