列车测试中的随机数据分裂函数sklearn返回错误

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：boolean，optional（default = True）是否要随机播放   拆分前的数据。如果shuffle = False则必须进行分层   无。

添加stratify=None，但我仍然遇到同样的错误。知道如何解决这个问题吗？

Answer 1

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

来检查当前安装的sklearn版本

>>> print(sklearn.__version__)