pbhatia243 / Neural_Conversation_Models

Tensorflow based Neural Conversation Models
Apache License 2.0
191 stars 73 forks source link

Logits scores of beam searched reply variants not available #8

Open bigdatabaracus opened 7 years ago

bigdatabaracus commented 7 years ago

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.