Bugfix: fixed sampling for labeled documents (credit: Xusong <song.xu@nlpr.ia.ac.cn>)