【问题标题】:How to set the History input of a RNN/LSTM cell to Zero?如何将 RNN/LSTM 单元的历史输入设置为零?
【发布时间】:2016-10-02 08:01:13
【问题描述】:

为了使用经过训练的 RNN/LSTM 模型对句子一一进行“推理”。我想我需要在每个句子的开头将RNN/LSTM单元的History input parameter设置为零。

那么在Tensorflow中怎么做呢?

【问题讨论】:

    标签: python machine-learning nlp tensorflow


    【解决方案1】:

    基于this answer,我相信状态会设置回您使用initial_state 参数传递给tf.nn.rnn(或其他RNN 创建函数)的任何状态。

    【讨论】:

    • 首先,谢谢。但是,我想我并不完全理解initial_state。我不确定这个设置是否同时影响 History 的参数和当前 Input 的参数。
    • 如果你看这里的代码,github.com/tensorflow/tensorflow/blob/master/tensorflow/python/…,LSTMCells 似乎没有历史参数。
    • 如果您的意思是 state 参数被调用时,则由 rnn 函数(或相关函数之一)处理。
    • 所以...这意味着每次我想推断一个句子时,我唯一应该做的就是设置initial_state,这意味着我不需要重新加载模型参数从检查点(仅第一次重新加载),对吗?
    • 我很确定您甚至不需要这样做。如果您将initial_state 参数设置为常数值为零,您应该能够让会话运行您的示例。
    猜你喜欢
    • 1970-01-01
    • 2020-11-23
    • 2018-03-06
    • 2017-10-25
    • 1970-01-01
    • 2018-03-19
    • 1970-01-01
    • 1970-01-01
    • 2018-07-06
    相关资源
    最近更新 更多