Modify the output of likelihood function in MlpBigram