方法“ train_test_split”中的参数“ stratify”(scikit学习)
我正在尝试train_test_split从scikit Learn软件包中使用,但是我在使用parameter时遇到了麻烦stratify。以下是代码: from sklearn import cross_validation, datasets X = iris.data[:,:2] y = iris.target cross_validation.train_test_split(X,y,stratify=y) 但是,我一直遇到以下问题: raise TypeError("Invalid parameters passed: %s" % str(options)) TypeError: Invalid parameters passed: {'stratify': array([0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, …