bug

[wxl112]

when the machine learning task is regression,it will comes up following questions! ValueError: The least populated class in y has only 1 member, which is too few. The minimum number of groups for any class cannot be less than 2.

[HiCiChlid]

I think 306 lines "train_features, valid_features, train_labels, valid_labels = train_test_split(features, labels, test_size = 0.15, stratify=labels)" is the final reason of the problem. stratify is for classification not for regression.