Neural_Conversation_Models icon indicating copy to clipboard operation
Neural_Conversation_Models copied to clipboard

Logits scores of beam searched reply variants not available

Open bigdatabaracus opened this issue 8 years ago • 0 comments

Hi, I'm trying to get access to the logit scores of the reply variants produced by the beam search. I was expecting the variable 'output_logits' produced by the model.step function in neural_conversation_model.py ( https://github.com/pbhatia243/Neural_Conversation_Models/blob/master/neural_conversation_model.py#L253 ) to contain this information, but this doesn't seem to be the case in the block where beamsearch is turned on.

bigdatabaracus avatar Dec 14 '16 15:12 bigdatabaracus