add calls to heldout_viterbi and heldout_state_marginals