fix degenerate issue in stop-or-not prediction