stratify by yi for train/test split
compare with svmlight