火车测试拆分功能中的随机数据返回错误

Question

import numpy as np
from sklearn.cross_validation import train_test_split as tts
a = np.array([[1,2,3,4,5]]).T
b = np.array([[100,200,50,60,3000]]).T
x1,x2,y1,y2 = tts(a,b, test_size=0.2, shuffle=False)
print(x1)
print(x2)
print(y1)
print(y2)

我总是收到以下错误：

builtins.TypeError: Invalid parameters passed: {'shuffle': False}

我已经检查了文档：

shuffle：布尔，可选（默认= true），无论是否洗牌 分裂之前的数据。如果shuffle = false，则必须分层 无。

添加了stratify=None，但是我仍然遇到同样的错误。任何想法如何解决？

Answer 1

这是由sklearn过时的版本引起的。该文档是指版本0.19.0。可以通过键入

来检查当前安装的sklearn的版本

>>> print(sklearn.__version__)