added heldout_viterbi method