WillKoehrsen / feature-selector

Feature selector is a tool for dimensionality reduction of machine learning datasets
GNU General Public License v3.0
2.23k stars 769 forks source link

bug #45

Open wxl112 opened 4 years ago

wxl112 commented 4 years ago

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 commented 2 years ago

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.